Fixes failing test.