fixed issue with running from cli