Update EndDisplayTest.java