Fixing a minor bug in one test