diff --git a/DESCRIPTION b/DESCRIPTION
index 2b0f092..cdb3fd2 100644
--- a/DESCRIPTION
+++ b/DESCRIPTION
@@ -23,7 +23,8 @@ Depends:
 Imports:
     data.table,
     keras,
-    rlang
+    rlang,
+    tensorflow
 Suggests:
     biomaRt,
     plotly,
diff --git a/R/neural.R b/R/neural.R
index f60354c..2a22769 100644
--- a/R/neural.R
+++ b/R/neural.R
@@ -5,7 +5,8 @@ neural <- function(preset, progress = NULL, seed = 49641) {
     reference_gene_ids <- preset$reference_gene_ids
 
     cached("neural", c(species_ids, gene_ids, reference_gene_ids), {
-        set.seed(seed)
+        tensorflow::set_random_seed(seed)
+
         gene_count <- length(gene_ids)
 
         progress_buffer <- 0