Fix code wrapping