Test fixes
0.5.1