changes to test code setup