Added a bit to make the script run fine when source is not 1