fix cli import error (for older bench)