Thanks Dmitry for providing a fix to the original code.
Signed-off-by: NEric Miao <eric.y.miao@gmail.com>
Signed-off-by: NDmitry Eremin-Solenikov <dbaryshkov@gmail.com>