changing test to reflect new layout