diff --git a/TP2/Armoire5.py b/TP2/Armoire5.py
index 6d70146d0860aadaff5b1c3ce2ee8a83f7e7c74b..74979443269082cf465e5a2f4c4cfe11ca1ed8b3 100644
--- a/TP2/Armoire5.py
+++ b/TP2/Armoire5.py
@@ -29,7 +29,7 @@ class Armoire5():
         self.debutAdrSource = 0
 
     def okCharge(self, i):
-        return 0 <= i <= 6
+        return 0 <= i <= self.nb - 1
 
     def resetAll(self):
         self.writeTOR(0)
diff --git a/TP2/Armoire6.py b/TP2/Armoire6.py
index 54e835fc424dc3659e58d4a5f63f1bd4b596b89a..bcaee0239b1418e104560b06fbb20a056d60ad14 100644
--- a/TP2/Armoire6.py
+++ b/TP2/Armoire6.py
@@ -32,7 +32,7 @@ class Armoire6():
         self.debutAdrSource=7
 
     def okCharge(self,i):
-        return 0 <= i <= 6
+        return 0 <= i <= nbCharge-1
     
     def resetAll(self):
         self.writeTOR(0)