Fix test case
Release 2.0.1