Fix for issue #2.

Fix for issue #2.