Fix import error for bjets module