diff --git a/config/config_test.yaml b/config/config_test.yaml
index 8d04935..90abc0f 100644
--- a/config/config_test.yaml
+++ b/config/config_test.yaml
@@ -18,4 +18,4 @@ training:
   batch_size: 64
   num_epochs: 1
   learning_rate: 0.0001
-  freeze_until_layer: classifier.1.weight
+  freeze_until_layer: classifier.1.0.weight