Fix 0.2 tests again