diff --git a/py/ttl_nn_latent_upscaler.py b/py/ttl_nn_latent_upscaler.py
index b3d737e..d7d69e2 100644
--- a/py/ttl_nn_latent_upscaler.py
+++ b/py/ttl_nn_latent_upscaler.py
@@ -310,4 +310,12 @@ class NNLatentUpscale:
         latent_out = latent_out.to(device="cpu")
 
         self.model.to(device=model_management.vae_offload_device())
-        return ({"samples": latent_out},)
\ No newline at end of file
+        return ({"samples": latent_out},)
+
+NODE_CLASS_MAPPINGS = {
+    "NNLatentUpscale": NNLatentUpscale
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "NNlLatentUpscale": "EFF-NN Latent Upscale"
+}