diff --git a/pl_bolts/callbacks/self_supervised.py b/pl_bolts/callbacks/self_supervised.py
index b6133620a0..5c133f10b0 100644
--- a/pl_bolts/callbacks/self_supervised.py
+++ b/pl_bolts/callbacks/self_supervised.py
@@ -103,7 +103,7 @@ def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx, data
         pl_module.logger.log_metrics(metrics, step=trainer.global_step)
 
 
-class BYOLMAWeightUpdate(pl.Callback):
+class BYOLMAWeightUpdate(Callback):
     """
     Weight update rule from BYOL.