Fixup tests
0.5.0