diff --git a/docs/docs.org b/docs/docs.org
index 240a547..2976d6d 100644
--- a/docs/docs.org
+++ b/docs/docs.org
@@ -120,7 +120,7 @@ class Disk(models.Model):
     size_in_gb = models.ForeignKey(DiskSize, on_delete=models.CASCADE)
 
     def __str__(self):
-        return '{} {} GB'.format(self.type, self.size)
+        return '{} {}'.format(self.type, self.size_in_gb)
 #+END_SRC
 
 ** Architecture, CpuManufacturer and Cpu
diff --git a/inventory/models.py b/inventory/models.py
index 27f1899..7178897 100644
--- a/inventory/models.py
+++ b/inventory/models.py
@@ -83,7 +83,7 @@ class Disk(models.Model):
     size_in_gb = models.ForeignKey(DiskSize, on_delete=models.CASCADE)
 
     def __str__(self):
-        return '{} {} GB'.format(self.type, self.size)
+        return '{} {}'.format(self.type, self.size_in_gb)
 
 
 class Architecture(models.Model):