Fix a bug with iteration space tests