Fixes issue #31. I love easy fixes :)