diff --git a/internal/provider/site_build_settings_resource.go b/internal/provider/site_build_settings_resource.go
index 39f5759..b651595 100644
--- a/internal/provider/site_build_settings_resource.go
+++ b/internal/provider/site_build_settings_resource.go
@@ -322,7 +322,10 @@ func (r *siteBuildSettingsResource) write(ctx context.Context, plan *siteBuildSe
 		},
 	}
 
-	if plan.BuildImage.IsUnknown() || plan.BuildImage.IsNull() {
+	if plan.BuildImage.IsUnknown() {
+		plan.BuildImage = curState.BuildImage
+	}
+	if plan.BuildImage.IsNull() {
 		site.BuildImage = curState.BuildImage.ValueStringPointer()
 	} else {
 		site.BuildImage = plan.BuildImage.ValueStringPointer()