Fix errors with missing class. See #64.