Fixing test failure