Fix for #203

Signed-off-by: ziycon <ziycon.eire@gmail.com>
Merge pull request #399 from ziycon/203

Fix #203