fix the main