fix for source mode tests