diff --git a/Makefile b/Makefile
index 246c9b39080b53c0a8cf801f79e9d1ccf0b36d54..864cfaf7c73cf26d9c50faafdaa507a9f2285c2f 100644
--- a/Makefile
+++ b/Makefile
@@ -22,11 +22,11 @@ uninstall: ## Uninstall
 	pip uninstall labml_nn
 
 docs: ## Render annotated HTML
-	python ../../pylit/pylit.py --remove_empty_sections --title_md -s ../../pylit/pylit_docs.css -t ../../pylit/template_docs.html -d html -w labml_nn
+	pylit --remove_empty_sections --title_md -t ../../pylit/templates/nn -d html -w labml_nn
 
 pages: ## Copy to lab-ml site
-	@cd ../lab-ml.github.io; git pull
-	cp -r html/* ../lab-ml.github.io/
+	@cd ../pages; git pull
+	cp -r html/* ../pages/
 
 help: ## Show this help.
 	@fgrep -h "##" $(MAKEFILE_LIST) | fgrep -v fgrep | sed -e 's/\\$$//' | sed -e 's/##//'
diff --git a/labml_nn/__init__.py b/labml_nn/__init__.py
index 84e78a71c64ce5cd38ab73d5ce1416bf34e65e9a..458595c1096415962d8fbecc98a0b7ac1dc6be3b 100644
--- a/labml_nn/__init__.py
+++ b/labml_nn/__init__.py
@@ -16,7 +16,7 @@ We will keep adding to this.
 contains implementations for
 [multi-headed attention](http://lab-ml.com/labml_nn/transformers/mha.html)
 and
-[relative multi-headed attention](http://lab-ml.com/labml_nn/transformers/relative_mha.html>).
+[relative multi-headed attention](http://lab-ml.com/labml_nn/transformers/relative_mha.html).
 
 #### ✨ [Recurrent Highway Networks](http://lab-ml.com/labml_nn/recurrent_highway_networks)
 
diff --git a/readme.md b/readme.md
index 4ea5ce8f8c475b69dbb9da23a252b08a5a88c7f9..fc3e5e9041d969a9ca6a88b582a77141f83e3461 100644
--- a/readme.md
+++ b/readme.md
@@ -15,7 +15,7 @@ We will keep adding to this.
 contains implementations for
 [multi-headed attention](http://lab-ml.com/labml_nn/transformers/mha.html)
 and
-[relative multi-headed attention](http://lab-ml.com/labml_nn/transformers/relative_mha.html>).
+[relative multi-headed attention](http://lab-ml.com/labml_nn/transformers/relative_mha.html).
 
 #### ✨ [Recurrent Highway Networks](http://lab-ml.com/labml_nn/recurrent_highway_networks)
 
diff --git a/requirements.txt b/requirements.txt
index e297706e59a2910aa8fadfb783ef7911605a4929..301ad41a8f62dee3b9e49de8449272cad8797918 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 torch>=1.6
-labml>=0.4.37
+labml>=0.4.52
 labml-helpers
 torchvision
 numpy>=1.16.3
diff --git a/setup.py b/setup.py
index 7eb1c547bd6ab48dafc9e05b6457ea2d7a0c3e96..606026fa604be7c85e78672f7679bc406df1cb00 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@ with open("readme.md", "r") as f:
 
 setuptools.setup(
     name='labml_nn',
-    version='0.4.10',
+    version='0.4.12',
     author="Varuna Jayasiri, Nipun Wijerathne",
     author_email="vpjayasiri@gmail.com, hnipun@gmail.com",
     description="A collection of PyTorch implementations of neural network architectures and layers.",
@@ -17,7 +17,7 @@ setuptools.setup(
     },
     packages=setuptools.find_packages(exclude=('test',
                                                'test.*')),
-    install_requires=['labml>=0.4.41',
+    install_requires=['labml>=0.4.52',
                       'labml_helpers',
                       'torch',
                       'einops',