Fix code container screen