fixing test_display test