Test fixes
0.5.2