diff --git a/R/mfairInitialization.R b/R/mfairInitialization.R
index 34e79b2..c732bd9 100644
--- a/R/mfairInitialization.R
+++ b/R/mfairInitialization.R
@@ -20,31 +20,29 @@ createMFAIR <- function(Y, X, Y_center = TRUE, K_max = 1L, project = "MFAIR") {
     stop("The number of samples in Y and X should be consistent!")
   } # End
 
-  if(is.matrix(Y)){
+  if (is.matrix(Y)) {
     Y_sparse <- FALSE
-  }else{
+  } else {
     Y_sparse <- TRUE
   }
 
   # Center the matrix Y
   if (Y_center) {
-    if(!Y_sparse){
+    if (!Y_sparse) {
       Y_mean <- mean(Y, na.rm = TRUE)
       Y <- Y - Y_mean
+    } else {
+      Y_mean <- mean(summary(Y)$x)
+      Y@x <- Y@x - Y_mean
     }
-else{
-  Y_mean <- mean(summary(Y)$x)
-  Y@x <- Y@x - Y_mean
-}
   } else {
     Y_mean <- 0
   }
 
   # Check Y's sparsity
-  if(!Y_sparse){
+  if (!Y_sparse) {
     n_missing <- sum(is.na(Y))
-  }
-  else{
+  } else {
     n_missing <- length(Y) - length(Y@x)
   }
   if (n_missing >= 1) {