fix classify

- closes #365