Small script bug fixed