diff --git a/cabal-ghc901.project b/cabal-ghc901.project
index 02f1519de0..5c233d5c8d 100644
--- a/cabal-ghc901.project
+++ b/cabal-ghc901.project
@@ -75,7 +75,7 @@ source-repository-package
 
 write-ghc-environment-files: never
 
-index-state: 2021-08-01T00:00:00Z
+index-state: 2021-08-12T12:00:38Z
 
 constraints:
     -- These plugins doesn't work on GHC9 yet
diff --git a/shake-bench/shake-bench.cabal b/shake-bench/shake-bench.cabal
index b966907cce..6dea31665f 100644
--- a/shake-bench/shake-bench.cabal
+++ b/shake-bench/shake-bench.cabal
@@ -20,7 +20,9 @@ library
         base == 4.*,
         Chart,
         Chart-diagrams,
-        diagrams,
+        diagrams-contrib,
+        diagrams-core,
+        diagrams-lib,
         diagrams-svg,
         directory,
         extra >= 1.7.2,