diff --git a/neat/common.py b/neat/common.py
index 7915276..42d2921 100644
--- a/neat/common.py
+++ b/neat/common.py
@@ -257,7 +257,7 @@ def calculate_migration_time(vms, bandwidth):
     """ Calculate the mean migration time from VM RAM usage data.
 
     :param vms: A map of VM UUIDs to the corresponding maximum RAM in MB.
-     :type vms: dict(str: int)
+     :type vms: dict(str: long)
 
     :param bandwidth: The network bandwidth in MB/s.
      :type bandwidth: float,>0
diff --git a/neat/locals/manager.py b/neat/locals/manager.py
index bf63edb..438bea9 100644
--- a/neat/locals/manager.py
+++ b/neat/locals/manager.py
@@ -409,7 +409,7 @@ def get_ram(vir_connection, vms):
      :type vms: list(str)
 
     :return: The maximum RAM for the VM UUIDs.
-     :rtype: dict(str : int)
+     :rtype: dict(str : long)
     """
     vms_ram = {}
     for uuid in vms:
@@ -431,7 +431,7 @@ def get_max_ram(vir_connection, uuid):
      :type uuid: str[36]
 
     :return: The maximum RAM of the VM in MB.
-     :rtype: int|None
+     :rtype: long|None
     """
     try:
         domain = vir_connection.lookupByUUIDString(uuid)