Fix for issue 1

Fix for #1.