fixed a couple of type errors
0.4.1