Fix Bok Choy tests
Fix broken Python unit test