diff --git a/Project.toml b/Project.toml
index 1d1eb4472..ac8973969 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,6 +1,6 @@
 name = "Metalhead"
 uuid = "dbeba491-748d-5e0e-a39e-b530a07fa0cc"
-version = "0.7.0"
+version = "0.7.1"
 
 [deps]
 Artifacts = "56f22d72-fd6d-98f1-02f0-08ddc0907c33"
diff --git a/src/convnets/densenet.jl b/src/convnets/densenet.jl
index 53d96df09..4269b83d2 100644
--- a/src/convnets/densenet.jl
+++ b/src/convnets/densenet.jl
@@ -147,9 +147,9 @@ Set `pretrain = true` to load the model with pre-trained weights for ImageNet.
 
 See also [`Metalhead.densenet`](#).
 """
-function DenseNet(config::Int = 121; pretrain = false)
+function DenseNet(config::Int = 121; pretrain = false, nclasses = 1000)
   @assert config in keys(densenet_config) "`config` must be one out of $(sort(collect(keys(densenet_config))))."
-  model = DenseNet(densenet_config[config])
+  model = DenseNet(densenet_config[config]; nclasses = nclasses)
 
   pretrain && loadpretrain!(model, string("DenseNet", config))
   return model