diff --git a/apps/base/tests/test_views.py b/apps/base/tests/test_views.py
index ed5a682..cab797d 100644
--- a/apps/base/tests/test_views.py
+++ b/apps/base/tests/test_views.py
@@ -28,15 +28,8 @@ class RootUnitTestCase(unittest.TestCase):
         real_hostname = socket.gethostname()
 
         for test_data in test_data_sets:
-            if 'hostname' in test_data:
-                hostname = test_data['hostname']
-            else:
-                hostname = real_hostname
-
-            if 'expected_color_hex' in test_data:
-                expected_color_hex = test_data['expected_color_hex']
-            else:
-                expected_color_hex = default_color_hex
+            hostname = test_data.get('hostname', real_hostname)
+            expected_color_hex = test_data.get('expected_color_hex', default_color_hex)
 
             with mock.patch('socket.gethostname', return_value=hostname):
                 view = RootView()