Fix unit test, refs #385