diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 98afbe54..a396f922 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -19,6 +19,7 @@ jobs:
       fail-fast: false
       matrix:
         version:
+          - '1.9'
           - '1'
         os:
           - ubuntu-latest
diff --git a/.github/workflows/ci_nightly.yml b/.github/workflows/ci_nightly.yml
deleted file mode 100644
index 34b96cf7..00000000
--- a/.github/workflows/ci_nightly.yml
+++ /dev/null
@@ -1,40 +0,0 @@
-name: CI (Julia nightly)
-on:
-  pull_request:
-    branches:
-      - dev
-  push:
-    branches:
-      - dev
-    tags: '*'
-jobs:
-  test-julia-nightly:
-    name: Julia ${{ matrix.version }} - ${{ matrix.os }} - ${{ matrix.arch }} - ${{ github.event_name }}
-    runs-on: ${{ matrix.os }}
-    strategy:
-      fail-fast: false
-      matrix:
-        version:
-          - 'nightly'
-        os:
-          - ubuntu-latest
-        arch:
-          - x64
-    steps:
-      - uses: actions/checkout@v3
-      - uses: julia-actions/setup-julia@v1
-        with:
-          version: ${{ matrix.version }}
-          arch: ${{ matrix.arch }}
-      - uses: actions/cache@v3
-        env:
-          cache-name: cache-artifacts
-        with:
-          path: ~/.julia/artifacts
-          key: ${{ runner.os }}-test-${{ env.cache-name }}-${{ hashFiles('**/Project.toml') }}
-          restore-keys: |
-            ${{ runner.os }}-test-${{ env.cache-name }}-
-            ${{ runner.os }}-test-
-            ${{ runner.os }}-
-      - uses: julia-actions/julia-buildpkg@v1
-      - uses: julia-actions/julia-runtest@v1
diff --git a/Project.toml b/Project.toml
index 589216d8..401403c6 100644
--- a/Project.toml
+++ b/Project.toml
@@ -21,9 +21,10 @@ ColorTypes = "0.10.3, 0.11"
 ComputationalResources = "0.3.2"
 Flux = "0.14"
 MLJModelInterface = "1.1.1"
-Metalhead = "0.9"
+Metalhead = "0.9.3"
 ProgressMeter = "1.7.1"
 StatisticalMeasures = "0.1"
+Statistics = "<0.0.1, 1"
 Tables = "1.0"
 julia = "1.9"
 
@@ -35,9 +36,8 @@ MLJBase = "a7f614a8-145f-11e9-1d2a-a57a1082229d"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
 StatisticalMeasures = "a19d573c-0a75-4610-95b3-7071388c7541"
-Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 StatsBase = "2913bbd2-ae8a-5f71-8c99-4fb6c76f3a91"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["CUDA", "cuDNN", "LinearAlgebra", "MLJBase", "Random", "StableRNGs", "StatisticalMeasures", "Statistics", "StatsBase", "Test"]
+test = ["CUDA", "cuDNN", "LinearAlgebra", "MLJBase", "Random", "StableRNGs", "StatisticalMeasures", "StatsBase", "Test"]
diff --git a/src/metalhead.jl b/src/metalhead.jl
index d6d53743..29b2b6fb 100644
--- a/src/metalhead.jl
+++ b/src/metalhead.jl
@@ -131,7 +131,7 @@ function VGGHack(
         "depth must be from one in $(sort(collect(keys(Metalhead.VGG_CONFIGS))))"
     )
     model = Metalhead.VGG(imsize;
-                config = Metalhead.VGG_CONV_CONFIGS[Metalhead.VGG_CONFIGS[depth]],
+                config = Metalhead.VGG_CONFIGS[depth],
                 inchannels,
                 batchnorm,
                 nclasses,