Fix code in different file formats