diff --git a/python/lammps.py b/python/lammps.py
index f23268cd60..bcf24a3fb1 100644
--- a/python/lammps.py
+++ b/python/lammps.py
@@ -390,10 +390,15 @@ class lammps(object):
   def extract_compute(self,id,style,type):
     if id: id = id.encode()
     if type == 0:
-      if style > 0: return None
-      self.lib.lammps_extract_compute.restype = POINTER(c_double)
-      ptr = self.lib.lammps_extract_compute(self.lmp,id,style,type)
-      return ptr[0]
+      if style == 0:
+        self.lib.lammps_extract_compute.restype = POINTER(c_double)
+        ptr = self.lib.lammps_extract_compute(self.lmp,id,style,type)
+        return ptr[0]
+      else if style == 1:
+        return None
+      else if style == 2:
+        self.lib.lammps_extract_compute.restype = POINTER(c_int)
+        return ptr[0]
     if type == 1:
       self.lib.lammps_extract_compute.restype = POINTER(c_double)
       ptr = self.lib.lammps_extract_compute(self.lmp,id,style,type)