Bug fix

Fixed class name