Updated unit test - refs #103