Fixed #4

Signed-off-by: hajee <hajee@moretIA.com>