Bug fix

Bug fix of the new script