Tests failed fix