fixes #10

Signed-off-by: haddada <haddadaseif@gmail.com>