Fix cli imports to allow proper package loading

Signed-off-by: Thanh Ha <thanh.ha@alumni.carleton.ca>