#620 fixed failing test

Resolves #620