Fixed test now that main method logic moved to run()