refactored test for proper class name