fix #4

Signed-off-by: Jayin Ton <tonjayin@gmail.com>