Fix test error on 2.0.0