Minor CLI fix, added missed file