diff --git a/soc/dos.f90 b/soc/dos.f90
index 87a88ef9..c9019bf5 100644
--- a/soc/dos.f90
+++ b/soc/dos.f90
@@ -38,7 +38,6 @@ subroutine dos_sub
    real(dp) :: k(3)
    real(dp) :: time_start, time_end, time_init
 
-   real(dp), allocatable :: kpoints(:, :)
    real(dp), allocatable :: eigval(:)
    real(dp), allocatable :: W(:)
    real(dp), allocatable :: omega(:)
@@ -67,7 +66,6 @@ subroutine dos_sub
    allocate(dos(NE))
    allocate(dos_mpi(NE))
    allocate(omega(NE))
-   kpoints= 0d0
    dos=0d0
    dos_mpi=0d0
    eigval= 0d0
diff --git a/soc/orbital_momenta.f90 b/soc/orbital_momenta.f90
index 2e9cfb52..6aaf6f3e 100644
--- a/soc/orbital_momenta.f90
+++ b/soc/orbital_momenta.f90
@@ -60,6 +60,7 @@ subroutine orbital_momenta(k, eigvec)
      sub= 0d0
      eig2= 0d0
 
+     if (Num_wann<7) return
 
      !> only for AlSb/InAs/GaSb/AlSb system
      !>> p orbital operators