diff --git a/Project.toml b/Project.toml
index 095a989..89131bd 100644
--- a/Project.toml
+++ b/Project.toml
@@ -48,8 +48,9 @@ Plots = "91a5bcdd-55d7-5caf-9e0b-520d859cae80"
 Tullio = "bc48ee85-29a4-5162-ae0b-a64e1601d4bc"
 cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd"
 
-[sources]
-NeuralClosure = {rev = "main", url = "https://github.com/DEEPDIP-project/NeuralClosure.jl.git"}
+[sources.NeuralClosure]
+rev = "main"
+url = "https://github.com/DEEPDIP-project/NeuralClosure.jl.git"
 
 [extensions]
 AttentionCNN = ["AttentionLayer"]
@@ -74,7 +75,7 @@ Documenter = "1"
 FFTW = "1"
 Glob = "1"
 IncompressibleNavierStokes = "3"
-JLD2 = "0.5"
+JLD2 = "0.5, 0.6"
 JuliaFormatter = "2"
 Juno = "0.8"
 Literate = "2"