Fix broken class ref
Merge pull request #11 from wmde/ref

Fix broken class ref