bug fix current

AshtonSBradley · Apr 3, 2023 · d4c806c · d4c806c
1 parent 20a1d9a
commit d4c806c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/analysis.jl b/src/analysis.jl
@@ -45,8 +45,8 @@ function current(psi::Psi{2},Ω = 0)
 	@unpack ψ,X = psi 
     x,y = X
     ψx,ψy = gradient(psi)
-	jx = @. imag(conj(ψ)*ψx) + Ω*y'  
-	jy = @. imag(conj(ψ)*ψy) - Ω*x
+	jx = @. imag(conj(ψ)*ψx) + Ω*abs2(ψ)*y'  
+	jy = @. imag(conj(ψ)*ψy) - Ω*abs2(ψ)*x 
 	return jx,jy
 end