quick fix to the script