Fix input
Fix main input