refactoring
Fixed wrong class name