Fix for issue #2.

Fix issue #2 from my personal account.