Fix a tests