Fix main() error