#52 fixed

I forgot to include this in last commit