Fix the 2 files