Fix command line to use new API