Fixing incorrect class name