small fix to the test