Fix #5
Signed-off-by: Alex <dtk077@gmail.com>