diff --git a/codegen.py b/codegen.py
index 14e2b98..ec5a23c 100644
--- a/codegen.py
+++ b/codegen.py
@@ -78,8 +78,8 @@ def read_code(code,save=True):
 	hero=data[0:6]
 	show=list(map(lambda x:x[0][x[1]],zip(tables.displays,hero)))
 	rank,cap=data[6:8]
-	if cap<4: cap=tables.cap[cap]
-	else: cap=rank
+	cap=tables.cap[cap]
+	if rank==12: cap=12
 	data=data[8:]
 	if rank<9: show[-1]='Hidden' # Don't spoil the talent
 	print(show)
diff --git a/tables.py b/tables.py
index c62c8ab..4f44a10 100644
--- a/tables.py
+++ b/tables.py
@@ -31,4 +31,4 @@ displays_nogun=[classes,armour,trait,spec,talent]
 displays_nogun_str=['classes','armour','traits','specs','talents']
 rank=range(12)
 xp=range(2500)
-cap=[3, 6, 9, 10, 11/12]
\ No newline at end of file
+cap=[3, 6, 9, 10, 11]
\ No newline at end of file