Fix for sourcing the script