Fix attempt on test_display