Fixed class name mismatch