Fix for Issue #4

Simple one line fix.