diff --git a/client/driver/fabfile.py b/client/driver/fabfile.py
index 4269590..838c8c3 100644
--- a/client/driver/fabfile.py
+++ b/client/driver/fabfile.py
@@ -808,22 +808,27 @@ def integration_tests():
 
     LOG.info("\n\nIntegration Tests: PASSED!!\n")
 
+    # Test task status UI
+    task_status_ui_test()
 
 @task
 def task_status_ui_test():
     # Test GPR
     upload_code = 'ottertuneTestTuningGPR'
     response = requests.get(dconf.WEBSITE_URL + '/test/task_status/' + upload_code)
+    LOG.info(get_content(response))
     assert 'Success:' in get_content(response)
 
     # Test DNN:
     upload_code = 'ottertuneTestTuningDNN'
     response = requests.get(dconf.WEBSITE_URL + '/test/task_status/' + upload_code)
+    LOG.info(get_content(response))
     assert 'Success:' in get_content(response)
 
     # Test DDPG:
     upload_code = 'ottertuneTestTuningDDPG'
     response = requests.get(dconf.WEBSITE_URL + '/test/task_status/' + upload_code)
+    LOG.info(get_content(response))
     assert 'Success:' in get_content(response)
 
     LOG.info("\n\nTask Status UI Tests: PASSED!!\n")
diff --git a/server/website/website/views.py b/server/website/website/views.py
index d13d266..7becae1 100644
--- a/server/website/website/views.py
+++ b/server/website/website/views.py
@@ -1681,7 +1681,7 @@ def tuner_status_test(request, upload_code):  # pylint: disable=unused-argument,
     context = _tuner_statue_helper(session.project.id, session.id, result.id)
     overall_status = context['overall_status']
     num_completed, num_total = context['num_completed'].replace(' ', '').split('/')
-    task_info = context['task_info']
+    task_info = context['tasks']
     num_tasks = len(task_info)
     if overall_status.lower() != 'success':
         return HttpResponse("Failure: overall status {} should be success".format(