small fix in the test