diff --git a/src/vpoller/agent.py b/src/vpoller/agent.py
index e827192..a237df3 100644
--- a/src/vpoller/agent.py
+++ b/src/vpoller/agent.py
@@ -1314,7 +1314,7 @@ def host_perf_counter_info(self, msg):
 
         try:
             metric_id = self.si.content.perfManager.QueryAvailablePerfMetric(entity=obj)
-        except pyVmomi.VmomiSupport.InvalidArgument as e:
+        except pyVmomi.vim.InvalidArgument as e:
             return {
                 'success': 1,
                 'msg': 'Cannot retrieve performance counters for %s: %s' % (msg['name'], e)
@@ -1566,7 +1566,7 @@ def vm_perf_counter_info(self, msg):
 
         try:
             metric_id = self.si.content.perfManager.QueryAvailablePerfMetric(entity=obj)
-        except pyVmomi.VmomiSupport.InvalidArgument as e:
+        except pyVmomi.vim.InvalidArgument as e:
             return {
                 'success': 1,
                 'msg': 'Cannot retrieve performance counters for %s: %s' % (msg['name'], e)
@@ -2478,7 +2478,7 @@ def datastore_perf_counter_info(self, msg):
 
         try:
             metric_id = self.si.content.perfManager.QueryAvailablePerfMetric(entity=obj)
-        except pyVmomi.VmomiSupport.InvalidArgument as e:
+        except pyVmomi.vim.InvalidArgument as e:
             return {
                 'success': 1,
                 'msg': 'Cannot retrieve performance counters for %s: %s' % (msg['name'], e)