fix the 2.0.1 dep