Quick fix!

Signed-off-by: Cydh Ramdh <house.bad@gmail.com>
Finish 1.0.1