fix a bug in the test added in prev commit