Fix code parsing