fix Issue5

This commit fixes #5.