diff --git a/Project.toml b/Project.toml
index 5e09b7b..102519c 100644
--- a/Project.toml
+++ b/Project.toml
@@ -24,6 +24,7 @@ JSON = "0.21, 1"
 Missings = "1"
 Reexport = "1.2"
 Statistics = "1"
+StatsBase = "0.34"
 julia = "1"
 
 [extras]
diff --git a/src/BigRiverQTL.jl b/src/BigRiverQTL.jl
index baf75e4..8de9148 100644
--- a/src/BigRiverQTL.jl
+++ b/src/BigRiverQTL.jl
@@ -2,7 +2,6 @@ module BigRiverQTL
     using BulkLMM
     using DataFrames, JSON, CSV
     using Statistics
-    using Distributed
     using LinearAlgebra
     import StatsBase: sample
 
diff --git a/test/kinship_test.jl b/test/kinship_test.jl
index 475f1e2..1a072b3 100644
--- a/test/kinship_test.jl
+++ b/test/kinship_test.jl
@@ -58,8 +58,8 @@ kinship_test(K1, "kinship_4way()")
 # Test: kinship_gs #
 ####################
 
-K1 = kinship_gs(geno1)
-K2 = kinship_gs(geno2)
+K1 = kinship_gs(geno1,0.9)
+K2 = kinship_gs(geno2,0.9)
 
 kinship_test(K1, "kinship_gs()")