test: Fix test to work with the latest CLI.

Signed-off-by: Blaž Hrastnik <blaz.hrast@gmail.com>