diff --git a/example_data/config.yaml b/example_data/config.yaml
index 7d908af..99d0a10 100644
--- a/example_data/config.yaml
+++ b/example_data/config.yaml
@@ -29,6 +29,7 @@ postprocess:
   top_k_per_sec: 200
 
 model:
+  name: UNetBackbone
   input_height: 128
   in_channels: 1
   out_channels: 32
diff --git a/src/batdetect2/models/backbones.py b/src/batdetect2/models/backbones.py
index 31feee0..07bc4b8 100644
--- a/src/batdetect2/models/backbones.py
+++ b/src/batdetect2/models/backbones.py
@@ -18,11 +18,10 @@ automatic padding to handle input sizes not perfectly divisible by the
 network's total downsampling factor.
 """
 
-from typing import Annotated, Literal, Tuple
+from typing import Annotated, Literal, Tuple, Union
 
 import torch
 import torch.nn.functional as F
-from example import Union
 from pydantic import Field
 from soundevent import data