diff --git a/denoising_diffusion_pytorch/karras_unet.py b/denoising_diffusion_pytorch/karras_unet.py
index 7db74e4ff..e73c663a4 100644
--- a/denoising_diffusion_pytorch/karras_unet.py
+++ b/denoising_diffusion_pytorch/karras_unet.py
@@ -371,7 +371,7 @@ def __init__(
         self.heads = heads
         hidden_dim = dim_head * heads
 
-        self.pixel_norm = PixelNorm(dim = 1)
+        self.pixel_norm = PixelNorm(dim = -1)
 
         self.attend = Attend(flash = flash)
 
diff --git a/denoising_diffusion_pytorch/version.py b/denoising_diffusion_pytorch/version.py
index 192e7e0f9..c283076a7 100644
--- a/denoising_diffusion_pytorch/version.py
+++ b/denoising_diffusion_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.10.8'
+__version__ = '1.10.9'