Fix for failing test ( not really broke )