diff --git a/sgdml/__init__.py b/sgdml/__init__.py
index 9cf0ae9..bfc2a61 100755
--- a/sgdml/__init__.py
+++ b/sgdml/__init__.py
@@ -22,7 +22,7 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
-__version__ = '0.5.0'
+__version__ = '0.5.1'
 
 MAX_PRINT_WIDTH = 100
 LOG_LEVELNAME_WIDTH = 7  # do not modify
diff --git a/sgdml/cli.py b/sgdml/cli.py
index 4b01f95..b4c8b1d 100644
--- a/sgdml/cli.py
+++ b/sgdml/cli.py
@@ -79,7 +79,7 @@ def _print_splash(max_memory, max_processes, use_torch):
 
     version_str = __version__
     version_str += (
-        ' ' + ui.yellow_back_str(' Latest: ' + latest_version + ' ')
+        ' ' + ui.color_str(' Latest: ' + latest_version + ' ', fore_color=ui.BLACK, back_color=ui.YELLOW, bold=True)
         if can_update
         else ''
     )
@@ -101,7 +101,7 @@ def _print_splash(max_memory, max_processes, use_torch):
     if can_update:
         print(
             '\n'
-            + ui.yellow_back_str(' UPDATE AVAILABLE ')
+            + ui.color_str(' UPDATE AVAILABLE ', fore_color=ui.BLACK, back_color=ui.YELLOW, bold=True)
             + '\n'
             + '-' * MAX_PRINT_WIDTH
         )