diff --git a/papyrus/measurements/__init__.py b/papyrus/measurements/__init__.py
index f842afd..0c19e37 100644
--- a/papyrus/measurements/__init__.py
+++ b/papyrus/measurements/__init__.py
@@ -23,6 +23,7 @@
 
 from papyrus.measurements.base_measurement import BaseMeasurement
 from papyrus.measurements.measurements import (
+    NTK,
     Accuracy,
     Loss,
     NTKEigenvalues,
@@ -41,4 +42,5 @@
     NTKMagnitudeDistribution.__name__,
     Loss.__name__,
     Accuracy.__name__,
+    NTK.__name__,
 ]