Fix main for deploy