I completely understand why they went the route they did. I'm just saying it would have been interesting the other way. Korra already had the elements mastered; she wouldn't be trying to master them again. She'd be trying to get them back, possibly by journeying into the spirit world.
Again, I know they weren't aware that they were getting a second season...and I get that it's a show for kids...I'm just saying it could have been another way to go.