From c19a6f6aacd3324ba41b413a01003a87b4d0420b Mon Sep 17 00:00:00 2001
From: markgpritchard <62439333+markgpritchard@users.noreply.github.com>
Date: Thu, 28 Mar 2024 12:17:31 +0000
Subject: [PATCH] Update TableOne.jl

---
 src/TableOne.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/TableOne.jl b/src/TableOne.jl
index 89abbc4..5848d3f 100644
--- a/src/TableOne.jl
+++ b/src/TableOne.jl
@@ -581,13 +581,13 @@ function _getvarname(var, varnames::Dict)
 end
 
 function _addnmissing!(_t, varvect, strataids)
-    n = _countmissing(varvect)
+    n = _countmissing(varvect, strataids)
     nmissing = [ "" for _ ∈ axes(_t, 1) ]
     nmissing[1] = sprint(show, n)
     insertcols!(_t, :nmissing => nmissing)
 end
 
-function _countmissing(varvect, sn=:Total)
+function _countmissing(varvect, strataids, sn=:Total)
     idmissing = findall(ismissing, varvect)
     vectorcountmissing = findall(x -> x ∈ idmissing, strataids[Symbol(sn)])
     n = length(vectorcountmissing)