diff --git a/algorithms/models/TransferNoise/model_TFN.m b/algorithms/models/TransferNoise/model_TFN.m
index 00959e5..6082ce2 100644
--- a/algorithms/models/TransferNoise/model_TFN.m
+++ b/algorithms/models/TransferNoise/model_TFN.m
@@ -1152,6 +1152,16 @@
             forcingData_colnames = forcingData_colnames(filt);
             setForcingData(obj, forcingData_data, forcingData_colnames);
 
+            % Update forcing data column numbers for those components using
+            % the raw inoput data. 
+            filt = find(strcmpi(varargin(:,2),'forcingdata'));
+            for ii=filt'
+               modelComponent = varargin{ii,1};
+               if ischar(varargin{ii,3})
+                  filt_forcingCols = cellfun(@(x,y)(strcmp(x,varargin{ii,3})), forcingData_colnames);
+                  obj.inputData.componentData.(modelComponent).dataColumn  = find(filt_forcingCols);
+               end
+            end
             %-------------
         end