modify logic to match the other tests