diff --git a/src/main/java/io/github/cottonmc/templates/model/TemplateModelVariantProvider.java b/src/main/java/io/github/cottonmc/templates/model/TemplateModelVariantProvider.java index 2e51f53..6bcf308 100644 --- a/src/main/java/io/github/cottonmc/templates/model/TemplateModelVariantProvider.java +++ b/src/main/java/io/github/cottonmc/templates/model/TemplateModelVariantProvider.java @@ -20,9 +20,7 @@ public class TemplateModelVariantProvider implements ModelVariantProvider { @Override public UnbakedModel loadModelVariant(ModelIdentifier modelId, ModelProviderContext context) throws ModelProviderException { - UnbakedModel variant = variants.get(modelId); - if (variant == null) throw new ModelProviderException("Couldn't find model for ID " + modelId); - return variant; + return variants.get(modelId); } public void registerTemplateModels(Block block, BlockState itemState, Function model) {