Fix source parameter so it will work when not specified