Fix issue #66

Good catch @tutengfei!