diff --git a/sdk/python/kubeflow/storage_init_container/hugging_face.py b/sdk/python/kubeflow/storage_init_container/hugging_face.py
index c136b265d2..b15297c4cf 100644
--- a/sdk/python/kubeflow/storage_init_container/hugging_face.py
+++ b/sdk/python/kubeflow/storage_init_container/hugging_face.py
@@ -36,7 +36,7 @@ class HuggingFaceModelParams:
 
     def __post_init__(self):
         # Custom checks or validations can be added here
-        if self.model_uri == "":
+        if self.model_uri == "" or self.model_uri is None:
             raise ValueError("model_uri cannot be empty.")
 
     @property
@@ -88,7 +88,7 @@ class HfDatasetParams:
 
     def __post_init__(self):
         # Custom checks or validations can be added here
-        if self.repo_id is None:
+        if self.repo_id == "" or self.repo_id is None:
             raise ValueError("repo_id is None")
 
     @property
diff --git a/sdk/python/kubeflow/storage_init_container/s3.py b/sdk/python/kubeflow/storage_init_container/s3.py
index cb5cb91cc0..5b1919644b 100644
--- a/sdk/python/kubeflow/storage_init_container/s3.py
+++ b/sdk/python/kubeflow/storage_init_container/s3.py
@@ -1,4 +1,4 @@
-from abstract_dataset_provider import datasetProvider
+from kubeflow.storage_init_container.abstract_dataset_provider import datasetProvider
 from dataclasses import dataclass, field
 import json, os
 import boto3