fixing test error