Fixes failing test