fix for issue #1

Simple fix.