Fixes test setup