Fix for #4

this seems to have fixed further errors for me