diff --git a/labml_nn/RWKV/experiment.py b/labml_nn/RWKV/experiment.py
index ad18fe3b..1f99d66d 100644
--- a/labml_nn/RWKV/experiment.py
+++ b/labml_nn/RWKV/experiment.py
@@ -5,8 +5,8 @@
 import torch.nn as nn
 from labml_nn.RWKV.configs import RWKVConfigs
 
-from __init__ import RWKV
-from __init__ import TimeMixing
+from labml_nn.RWKV import RWKV
+from labml_nn.RWKV import TimeMixing
 from labml import experiment
 from labml.configs import option
 from labml_nn.experiments.nlp_autoregression import NLPAutoRegressionConfigs