Fixed kbe_unittest.py