Fixed test. refs #398