Fix broken test setup