Edit test for the new layout