Fixing

Signed-off-by: feafarot <sly.feafarot@gmail.com>
Fixed script

Signed-off-by: feafarot <sly.feafarot@gmail.com>