fixed minor error with the test