diff --git a/R/gbm.step.R b/R/gbm.step.R
index 18dfc1b..a11c42b 100644
--- a/R/gbm.step.R
+++ b/R/gbm.step.R
@@ -118,7 +118,7 @@ gbm.step <- function (
 	if (is.null(fold.vector)) {
 
 		if (prev.stratify & family == "bernoulli") {
-			presence.mask <- data[,gbm.y] == 1
+			presence.mask <- data[,gbm.y] > 0
 			absence.mask <- data[,gbm.y] == 0
 			n.pres <- sum(presence.mask)
 			n.abs <- sum(absence.mask)
diff --git a/R/gbm.utils.R b/R/gbm.utils.R
index a17cca5..ede396a 100644
--- a/R/gbm.utils.R
+++ b/R/gbm.utils.R
@@ -22,8 +22,8 @@
         stop("obs and preds must be equal lengths")
 	}
     n.x <- length(obsdat[obsdat == 0])
-    n.y <- length(obsdat[obsdat == 1])
-    xy <- c(preddat[obsdat == 0], preddat[obsdat == 1])
+    n.y <- length(obsdat[obsdat > 0])
+    xy <- c(preddat[obsdat == 0], preddat[obsdat > 0])
     rnk <- rank(xy)
     wilc <- ((n.x * n.y) + ((n.x * (n.x + 1))/2) - sum(rnk[1:n.x]))/(n.x * n.y)
     return(round(wilc, 4))