Possible fix for Issue #50

I was not able to replicate the same error after this small change