diff --git a/grits/coarsegrain.py b/grits/coarsegrain.py index 28e3d72..8ff2847 100644 --- a/grits/coarsegrain.py +++ b/grits/coarsegrain.py @@ -362,7 +362,7 @@ def visualize( if atom_names: atomistic.save( os.path.join(tmp_dir, "atomistic_tmp.mol2"), - show_ports=show_ports, + include_ports=show_ports, overwrite=True, ) @@ -397,7 +397,7 @@ def visualize( simplefilter("ignore") coarse.save( os.path.join(tmp_dir, "coarse_tmp.mol2"), - show_ports=show_ports, + include_ports=show_ports, overwrite=True, ) with open(os.path.join(tmp_dir, "coarse_tmp.mol2"), "r") as f: