Fix

I did not check if this code works, so...
Merge pull request #3 from masasron/patch-1

Fix