fix to cli tests