Fixed Issue #2

Thanks, @vtforester !