diff --git a/modules/deeplearning/cloud_fraction_fcn_abi.py b/modules/deeplearning/cloud_fraction_fcn_abi.py
index f57b79f84c5016668aa4df9dc7d1eed9688c8e96..014210e33df560d515cdf49d42946e9c2817796a 100644
--- a/modules/deeplearning/cloud_fraction_fcn_abi.py
+++ b/modules/deeplearning/cloud_fraction_fcn_abi.py
@@ -1,7 +1,7 @@
 import tensorflow as tf
 
 from util.plot_cm import confusion_matrix_values
-from util.setup import logdir, modeldir, now, ancillary_path
+from util.setup_cloud_fraction import logdir, modeldir, now, ancillary_path
 from util.util import EarlyStop, normalize, denormalize, get_grid_values_all
 import glob
 import os, datetime
@@ -806,7 +806,8 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
     if out_file is not None:
         np.save(out_file, (cld_frac_out, bt, refl, cp, lons, lats))
     else:
-        return [cld_frac_out, bt, refl, cp, lons, lats]
+        # return [cld_frac_out, bt, refl, cp, lons, lats]
+        return cld_frac_out
 
 
 def run_evaluate_static_full_disk(in_file, out_file, ckpt_dir):
@@ -862,7 +863,8 @@ def run_evaluate_static_full_disk(in_file, out_file, ckpt_dir):
     if out_file is not None:
         np.save(out_file, (cld_frac_out, bt, refl, cp, lons, lats))
     else:
-        return [cld_frac_out, bt, refl, cp, lons, lats]
+        # return [cld_frac_out, bt, refl, cp, lons, lats]
+        return cld_frac_out
 
 
 def run_evaluate_static_valid(in_file, out_file, ckpt_dir):