Fixes test failure