Fixed last failing test