Fix code layout bug