diff --git a/src/layers/stateless.jl b/src/layers/stateless.jl
index cc1be044..23fd1651 100644
--- a/src/layers/stateless.jl
+++ b/src/layers/stateless.jl
@@ -50,7 +50,7 @@ function normalise(x::AbstractArray; dims=1)
   return (x .- μ′) ./ σ′
 end
 
-function normalise(x::AbstractArray, dims=1)
+function normalise(x::AbstractArray, dims)
   Base.depwarn("`normalise(x::AbstractArray, dims)` is deprecated, use `normalise(a, dims=dims)` instead.", :normalise)
   normalise(x, dims = dims)
 end