Adds the correct environment variable for diego staging

[#82172176]