Add test for the expand display