and fixing the test case to display correctly as well