diff --git a/src/batdetect2/cli/train.py b/src/batdetect2/cli/train.py
index b16a587..e574b38 100644
--- a/src/batdetect2/cli/train.py
+++ b/src/batdetect2/cli/train.py
@@ -70,7 +70,7 @@ def train_command(
         val_annotations = load_dataset_from_config(val_dataset)
         logger.debug(
             "Loaded {num_annotations} validation examples",
-            num_files=len(val_annotations),
+            num_annotations=len(val_annotations),
         )
     else:
         logger.debug("No validation directory provided.")
diff --git a/src/batdetect2/models/bottleneck.py b/src/batdetect2/models/bottleneck.py
index 7a38a11..c72f66a 100644
--- a/src/batdetect2/models/bottleneck.py
+++ b/src/batdetect2/models/bottleneck.py
@@ -22,7 +22,6 @@ from torch import nn
 
 from batdetect2.configs import BaseConfig
 from batdetect2.models.blocks import (
-    LayerConfig,
     SelfAttentionConfig,
     VerticalConv,
     build_layer_from_config,