fixing test failure