tweak test setup