diff --git a/NAMESPACE b/NAMESPACE
index 96a0d5dfd..a3c6f5008 100644
--- a/NAMESPACE
+++ b/NAMESPACE
@@ -89,6 +89,7 @@ export(vaeac_plot_eval_crit)
 export(vaeac_plot_imputed_ggpairs)
 export(vaeac_train_model)
 export(vaeac_train_model_continue)
+export(weight_matrix)
 export(weight_matrix_cpp)
 importFrom(Rcpp,sourceCpp)
 importFrom(data.table,":=")
diff --git a/R/setup_computation.R b/R/setup_computation.R
index 6287ec633..f4f3f60f0 100644
--- a/R/setup_computation.R
+++ b/R/setup_computation.R
@@ -662,6 +662,7 @@ feature_group_not_exact <- function(group_num, n_combinations = 200, weight_zero
 #' @return Numeric matrix. See [weight_matrix_cpp()] for more information.
 #' @keywords internal
 #'
+#' @export
 #' @author Nikolai Sellereite, Martin Jullum
 weight_matrix <- function(X, normalize_W_weights = TRUE, is_groupwise = FALSE) {
   # Fetch weights