diff --git a/docs/Project.toml b/docs/Project.toml
index 6e418099..5adb1755 100644
--- a/docs/Project.toml
+++ b/docs/Project.toml
@@ -4,4 +4,4 @@ JustRelax = "34418575-392d-4e26-8c6d-96b0910afa06"
 
 [compat]
 Documenter = "1"
-JustRelax = "0.2"
+JustRelax = "0.2, 0.3"
diff --git a/src/common.jl b/src/common.jl
index d32e5a9a..1adf2dfb 100644
--- a/src/common.jl
+++ b/src/common.jl
@@ -13,6 +13,7 @@ export @allocate,
     @idx,
     @copy,
     @velocity,
+    @displacement,
     @strain,
     @stress,
     @tensor,
diff --git a/test/Project.toml b/test/Project.toml
index aedc62d2..79312fc9 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -22,9 +22,9 @@ AllocCheck = "0.1"
 CUDA = "~5.3.5"
 CellArrays = "0.2"
 GeoParams = "0.5, 0.6"
-JustPIC = "0.3.4, 0.3.5, 0.4"
+JustPIC = "0.3.4, 0.3.5, 0.3.6, 0.4"
 MPI = "0.20"
-ParallelStencil = "0.11, 0.12, 0.13"
+ParallelStencil = "0.11, 0.12"
 StaticArrays = "1"
 Statistics = "1"
 Suppressor = "0.2"