updating unit helpers
adding missed variable