Fixed name of class
See #2