fixing test bug