updated test to accomodate layout change