remove test class import