Fix unit tests for FlxAtlas API changes