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

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