Bug fix for Issue #22

Very small fix.