fix test cases
Release 0.5.2