Fix Python tests
Fix Bok Choy tests