Fix a testing issue