Fixing the custom management command for cli