Add the highlighted nodes shown in the images below, location at the top of the screenshots. Making sure that your enabled variables on the nodes match the screenshots.