diff --git a/docs/source/self_supervised_models.rst b/docs/source/self_supervised_models.rst
index 9296a64b0d..bf956c73ba 100644
--- a/docs/source/self_supervised_models.rst
+++ b/docs/source/self_supervised_models.rst
@@ -431,7 +431,7 @@ To Train::
     model = SwAV(
         gpus=1,
         num_samples=dm.num_unlabeled_samples,
-        dataset='stl10,
+        dataset='stl10',
         batch_size=batch_size
     )