Explicitly set the class property. Fixes #36 Thanks @rewchelsea