Fixed #1

Signed-off-by: Samuel <samahl0916@gmail.com>