Fix code that got cut off.