diff --git a/cbutil/__init__.py b/cbutil/__init__.py
index 6bd35ff8c90142a633f1794ee3c69c1fc71134c4..91c8e0a270f9299a76db9dac1aa60e487277f1b3 100644
--- a/cbutil/__init__.py
+++ b/cbutil/__init__.py
@@ -5,3 +5,4 @@ from .util import read_file_line_wise, time_conversion, file_time_to_sec, get_fr
 from .data_points import DataPoint, data_point_factory
 from .get_job_info import get_url_from_env, get_job_datapoints
 from .gitlab_api import get_git_infos_from_api
+from .update_data import get_updated_data
diff --git a/cbutil/update_data.py b/cbutil/update_data.py
index 32751917549f5ba7005b467bf0fd8aa27f35dbb7..753245e9e3cfae07c72f0e3cd5524f576a33643d 100644
--- a/cbutil/update_data.py
+++ b/cbutil/update_data.py
@@ -37,10 +37,10 @@ def apply_updates(old_data, update_mapping):
     """
     for data_point in old_data:
         for update_key, mapping in update_mapping.items():
-            if (current_value := data_point.get(update_key)):
-                for old_value, new_value in mapping.items():
-                    if current_value == old_value:
-                        data_point[update_key] = new_value
+            current_value = data_point.get(update_key, None)
+            for old_value, new_value in mapping.items():
+                if current_value == old_value:
+                    data_point[update_key] = new_value
 
         yield data_point