fixed #10, thanks to @laoyang945