Fix type signature in main.