Fix/failing test