fix the test. Closes #406