Sean/fix 604

Fixes #604