resolve #5

Signed-off-by: David Lu <dayal29699@gmail.com>