Fix for issue 1

fixes issue #1