trying to fix test setup