tweak test code