diff --git a/model_angelo/__init__.py b/model_angelo/__init__.py
index 3a884dc..1a5910d 100644
--- a/model_angelo/__init__.py
+++ b/model_angelo/__init__.py
@@ -5,4 +5,4 @@
 """
 
 
-__version__ = "1.0.1"
+__version__ = "1.0.2"
diff --git a/model_angelo/gnn/inference.py b/model_angelo/gnn/inference.py
index 274ce30..c77d8bd 100644
--- a/model_angelo/gnn/inference.py
+++ b/model_angelo/gnn/inference.py
@@ -57,7 +57,7 @@ def infer(args):
     batch_converter = alphabet.get_batch_converter()
 
     lang_model = lang_model.eval()
-    voxel_size = args.voxel_size
+    voxel_size = getattr(args, "voxel_size", 1.0)
 
     rna_sequences, dna_sequences = [], []
     if args.rna_fasta is not None:
diff --git a/model_angelo/gnn/inference_no_seq.py b/model_angelo/gnn/inference_no_seq.py
index ff12d6d..ac43cf1 100644
--- a/model_angelo/gnn/inference_no_seq.py
+++ b/model_angelo/gnn/inference_no_seq.py
@@ -42,7 +42,7 @@ def infer(args):
     device_names = get_device_names(args.device)
     num_devices = len(device_names)
 
-    voxel_size = args.voxel_size
+    voxel_size = getattr(args, "voxel_size", 1.0)
 
     protein = None
     if args.struct.endswith("prot"):