Fixed toggle between test and class file. (Broken when changing FileCreator API)