diff --git a/SciFiReaders/readers/microscopy/spm/afm/gwyddion.py b/SciFiReaders/readers/microscopy/spm/afm/gwyddion.py
index 6839160..cee4ec4 100644
--- a/SciFiReaders/readers/microscopy/spm/afm/gwyddion.py
+++ b/SciFiReaders/readers/microscopy/spm/afm/gwyddion.py
@@ -125,18 +125,31 @@ def gsf_read(self):
         data_set.data_type = 'Image'
 
         #Add quantity and units
-        data_set.units = "a.u" #metadata['ZUnits']
+        
+        try:
+            data_set.units = metadata['ZUnits']
+        except Exception as e:
+            # If an error occurs, fall back to using "a.u"
+            data_set.units = "a.u"
         data_set.quantity = metadata['Title']
-       
+
         #Add dimension info
         data_set.set_dimension(0, sid.Dimension(np.linspace(0, metadata['XReal'], num_cols),
                                                 name = 'x',
-                                                units= "a.u", #metadata['XYUnits'], 
+                                                try:
+                                                    units = metadata['XYUnits'],
+                                                except Exception as e:
+                                                    # If an error occurs, fall back to using "a.u"
+                                                    units= "a.u",  
                                                 quantity = 'x',
                                                 dimension_type='spatial'))
         data_set.set_dimension(1, sid.Dimension(np.linspace(0, metadata['YReal'], num_rows),
                                                 name = 'y',
-                                                units="a.u", #metadata['XYUnits'], 
+                                                try:
+                                                    units = metadata['XYUnits'], 
+                                                except Exception as e:
+                                                    # If an error occurs, fall back to using "a.u"
+                                                    units= "a.u",
                                                 quantity = 'y',
                                                 dimension_type='spatial'))