fixing class name for tests