Fix running from the command line.