change test class to the wright place