Fix Issue #1

Thanks @zenggang for reporting the bug.