Fixing test setup