Fixing test error and test class name