fix failing test from last commit