diff --git a/src/fiats/neural_network_m.f90 b/src/fiats/neural_network_m.f90
index ce99c9b02..0b78067dd 100644
--- a/src/fiats/neural_network_m.f90
+++ b/src/fiats/neural_network_m.f90
@@ -21,7 +21,7 @@ module neural_network_m
   type neural_network_t(k)
     !! Encapsulate the information needed to perform inference
     integer, kind :: k = default_real 
-    type(tensor_map_t(k)) :: input_map_, output_map_
+    type(tensor_map_t(k)), private :: input_map_, output_map_
     type(metadata_t), private :: metadata_
     real(k), allocatable, private :: weights_(:,:,:), biases_(:,:)
     integer, allocatable, private :: nodes_(:)
diff --git a/src/fiats/trainable_network_s.f90 b/src/fiats/trainable_network_s.f90
index 7c2f8a909..7b41843b4 100644
--- a/src/fiats/trainable_network_s.f90
+++ b/src/fiats/trainable_network_s.f90
@@ -18,8 +18,9 @@
   module procedure default_real_map_to_training_ranges
     associate(inputs => input_output_pair%inputs(), expected_outputs => input_output_pair%expected_outputs())
       associate( &
-         normalized_inputs  => self%input_map_%map_to_training_range(inputs), &
-         normalized_outputs => self%output_map_%map_to_training_range(expected_outputs) &
+         normalized_inputs  => self%map_to_input_range(inputs), &
+         normalized_outputs => self%map_from_output_range(expected_outputs) &
+
       )
         normalized_input_output_pair = input_output_pair_t(normalized_inputs, normalized_outputs)
       end associate