diff --git a/TicTacToe_AI/Net/pytorch_ai.py b/TicTacToe_AI/Net/pytorch_ai.py
index 3d0f6a3..0a4dbf0 100644
--- a/TicTacToe_AI/Net/pytorch_ai.py
+++ b/TicTacToe_AI/Net/pytorch_ai.py
@@ -90,6 +90,7 @@ def buildsets():
     return trainset, testset
 
 
+# TODO testnet is wrong since batching has been implemented
 def testnet(net, testset, device):
     correct = 0
     total = 0