fixing test code