FIX: fixing the script