fix classify

- closes #365
Merge pull request #366 from epeli/fix-classify

fix classify