diff --git a/kfac.py b/kfac.py
index 0f186b7..f15e6a6 100644
--- a/kfac.py
+++ b/kfac.py
@@ -5,7 +5,7 @@ import torch.optim as optim
 
 
 # TODO: In order to make this code faster:
-# 1) Implement _extract_patches as a single cuda kernel_size
+# 1) Implement _extract_patches as a single cuda kernel
 # 2) Compute QR decomposition in a separate process
 # 3) Actually make a general KFAC optimizer so it fits PyTorch