Fix unit test
Previous commit: closes #619.

This commit fixes unit test, oops.