diff --git a/examples/transformer.py b/examples/transformer.py
index 855dfa20fe..7bc2157e9f 100755
--- a/examples/transformer.py
+++ b/examples/transformer.py
@@ -2,7 +2,8 @@
 import numpy as np
 import random
 
-from tinygrad.nn.optim import Adam, get_parameters
+from tinygrad.state import get_parameters
+from tinygrad.nn.optim import Adam
 from extra.training import train, evaluate
 from models.transformer import Transformer