Tests : mock fix