Fix bok choy test with a wait and element finder changes.