Add check in main.