fixed the test setup!