Fix for issue-1

This fixes #1.