fixes test code