Fix bok_choy tests