Fix 2.1.0 testing