Fix 2.2.1 error