Fix issue #16

Fixes issue #16 with weird class name.