Small update to fix failing test