Fix input format

Test Plan: