From 3de3086a62e9c2e8543bcbd40d7da992827709ea Mon Sep 17 00:00:00 2001
From: pepijnvink <pepijnvink@gmail.com>
Date: Thu, 14 Dec 2023 13:10:52 +0100
Subject: [PATCH] group by .imp if group=NULL

---
 R/ggmice.R | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/R/ggmice.R b/R/ggmice.R
index 70b9b6a0..b1acc6d2 100644
--- a/R/ggmice.R
+++ b/R/ggmice.R
@@ -111,6 +111,10 @@ ggmice <- function(data = NULL,
     )
     mice_mapping <-
       utils::modifyList(mapping, ggplot2::aes(colour = .where))
+    if (is.null(mapping$group)) {
+      mice_mapping <-
+        utils::modifyList(mice_mapping, ggplot2::aes(group = .imp))
+    }
     mice_colors <-
       c("observed" = "#006CC2B3",
         "imputed" = "#B61A51B3")