first stab at updating tests for ux overhaul