Fix for failing test again