diff --git a/include/epanet_py.h b/include/epanet_py.h
index 3722731..202d97d 100644
--- a/include/epanet_py.h
+++ b/include/epanet_py.h
@@ -64,7 +64,7 @@ int DLLEXPORT rprt_anlysstats(Handle ph, EN_AnalysisStatistic code, EN_API_FLOAT
 
 int DLLEXPORT anlys_getoption(Handle ph, EN_Option opt, EN_API_FLOAT_TYPE *value);
 int DLLEXPORT anlys_setoption(Handle ph, int code, EN_API_FLOAT_TYPE value);
-int DLLEXPORT anlys_getflowunits(Handle ph, EN_FlowUnits *code);
+int DLLEXPORT anlys_getflowunits(Handle ph, int *code);
 int DLLEXPORT anlys_setflowunits(Handle ph, EN_FlowUnits code);
 int DLLEXPORT anlys_gettimeparam(Handle ph, EN_TimeProperty code, long *value);
 int DLLEXPORT anlys_settimeparam(Handle ph, EN_TimeProperty code, long value);
diff --git a/src/epanet_py.c b/src/epanet_py.c
index 394df5f..6757340 100644
--- a/src/epanet_py.c
+++ b/src/epanet_py.c
@@ -246,10 +246,10 @@ int DLLEXPORT anlys_setoption(Handle ph, EN_Option code, EN_API_FLOAT_TYPE value
     return error_set(pr->error, EN_setoption(pr->project, code, value));
 }
 
-int DLLEXPORT anlys_getflowunits(Handle ph, EN_FlowUnits *code)
+int DLLEXPORT anlys_getflowunits(Handle ph, int *code)
 {
     handle_t *pr = (handle_t *)ph;
-    return error_set(pr->error, EN_getflowunits(pr->project, (int *)code));
+    return error_set(pr->error, EN_getflowunits(pr->project, code));
 }
 
 int DLLEXPORT anlys_setflowunits(Handle ph, EN_FlowUnits code)