diff --git a/src/schroed_dirac_solver.f90 b/src/schroed_dirac_solver.f90
index 376a4ef..00b8256 100644
--- a/src/schroed_dirac_solver.f90
+++ b/src/schroed_dirac_solver.f90
@@ -64,6 +64,7 @@ subroutine total_energy(Z, rmax, Ne, a, p, Nq, DOFs, alpha_int, dirac_int, &
 
     allocate(H(DOFS, DOFS), S(DOFS), DSQ(DOFS), uq(Nq,Ne))
     allocate(D(DOFS, DOFS), lam2(DOFS), rho(Nq,Ne), fullc(Nn))
+    allocate(lam_tmp(DOFS))
     if (dirac_int == 1) then
         allocate(lam(47))
         allocate(eigfn(Nq, Ne, 49))