diff --git a/src/ActiveInference.jl b/src/ActiveInference.jl
index 6e9c699..a97eee4 100644
--- a/src/ActiveInference.jl
+++ b/src/ActiveInference.jl
@@ -1,3 +1,6 @@
+using LinearAlgebra
+using Plots
+
 module ActiveInference
 
 include("functions.jl")