Fix a failling test