diff --git a/src/nf/nf_network_submodule.f90 b/src/nf/nf_network_submodule.f90
index df4e04cc..0071e4ef 100644
--- a/src/nf/nf_network_submodule.f90
+++ b/src/nf/nf_network_submodule.f90
@@ -369,11 +369,11 @@ module subroutine print_info(self)
   end subroutine print_info
 
 
-  pure module function get_num_params(self) result(num_params)
+  pure module function get_num_params(self)
     class(network), intent(in) :: self
-    integer :: num_params
+    integer :: get_num_params
 
-    num_params = sum(self % layers % get_num_params())
+    get_num_params = sum(self % layers % get_num_params())
 
   end function get_num_params