Fix minor script bug