diff --git a/R/get_probs.R b/R/get_probs.R
index ac04fed..4086426 100644
--- a/R/get_probs.R
+++ b/R/get_probs.R
@@ -34,6 +34,7 @@ get_initial_probs.nhmm <- function(model, probs, ...) {
   }
   if (attr(model$X_pi, "icpt_only")) {
     X <- model$X_pi[, 1L, drop = FALSE]
+    ids <- "all"
   } else {
     X <- model$X_pi
   }
@@ -104,6 +105,7 @@ get_transition_probs.nhmm <- function(model, probs, remove_voids = TRUE, ...) {
   }
   if (!attr(model$X_A, "iv")) {
     X <- model$X_A[, , 1L, drop = FALSE]
+    ids <- "all"
   } else {
     X <- model$X_A
   }
@@ -192,6 +194,7 @@ get_emission_probs.nhmm <- function(model, probs, remove_voids = TRUE, ...) {
   }
   if (!attr(model$X_B, "iv")) {
     X <- model$X_B[, , 1L, drop = FALSE]
+    ids <- "all"
   } else {
     X <- model$X_B
   }
@@ -286,6 +289,7 @@ get_cluster_probs.mnhmm <- function(model, probs, ...) {
   }
   if (attr(model$X_omega, "icpt_only")) {
     X <- model$X_omega[, 1L, drop = FALSE]
+    ids <- "all"
   } else {
     X <- model$X_omega
   }