diff --git a/test/test_workers.py b/test/test_workers.py
index d7029a2..381c898 100644
--- a/test/test_workers.py
+++ b/test/test_workers.py
@@ -919,7 +919,6 @@ def test_worker_assign_as_block_preference2():
     """Test that a worker's assign_as_block preference is respected over the global setting."""
     Rota = generate_basic_rota(workers=4, weeks_to_rota=8)
 
-    # Worker 1 prefers to have shift 'a' assigned as a block, worker 2 does not
     workers = Rota.get_workers()
     worker1 = workers[0]
     worker2 = workers[1]
@@ -952,10 +951,12 @@ def test_worker_assign_as_block_preference2():
     a_shift.assign_as_block = True  # Set the shift 'a' to be assigned as a block globally
     Rota.build_and_solve(options={"ratio": 0.0})
 
+    Rota.export_rota_to_html("test_worker_assign_as_block_preference", folder="tests")
+
     for worker in Rota.get_workers():
         shift_list = Rota.get_worker_shift_list_string(worker)
         # All workers should have 'a' shifts grouped together (block)
-        assert shift_list.count("aaaa") == 3, f"Worker {worker.name} should have all 'a' shifts grouped together (block)"
+        assert shift_list.count("aaaa") == 2, f"Worker {worker.name} should have all 'a' shifts grouped together (block)"
 
 def test_worker_assign_as_block_preference_multiple_shifts():
     """Test that a worker's assign_as_block preference is respected over the global setting."""