Fix test failure
0.5.4