Sean/fix 604

Fixes 604