diff --git a/lib/BNNS/BNNS.jl b/lib/BNNS/BNNS.jl
index 07aba49..34fbb43 100644
--- a/lib/BNNS/BNNS.jl
+++ b/lib/BNNS/BNNS.jl
@@ -42,4 +42,4 @@ function rand! end
 function randn! end
 function seed! end
 
-end
\ No newline at end of file
+end
diff --git a/test/BNNS.jl b/test/BNNS.jl
index e6ce6cc..6d31958 100644
--- a/test/BNNS.jl
+++ b/test/BNNS.jl
@@ -1,3 +1,4 @@
+using .BNNS
 const RAND_TYPES = [BFloat16, Float16, Float32, Int8, UInt8, Int16, UInt16, Int32, UInt32, Int64,
                     UInt64]
 const RANDN_TYPES = [BFloat16, Float16, Float32]