After taking an early lead through Balotelli, Massimiliano Allegri's side found themselves 2-1 down heading into the final stages of Saturday's game following goals from Luca Siligardi and Paulinho.
Up stepped Balotelli to earn a share of the spoils with a fine set-piece six minutes from time and the 23-year-old called on his team-mates to focus on concentrating in order to pick up maximum points.
"We need to be more careful when our opponents go on the counter-attack, because every time they do we tend to concede," he told Sky Sport Italia.
"We are playing fairly well and are scoring again at last, which was a problem before, so now we just need a little more luck."
Balotelli's double made it four goals in three games and he was quick to address reports he previously had fitness problems after going six games without a goal for Milan.
"I have always been in good shape and never had any of the problems that were written about," he added. "I just wasn't scoring, that's all."
‘Even though it was late in the game, the Arsenal fans started singing my name. It felt like the moment I announced myself as a professional footballer’: Ex-Gunners star recalls scoring first senior goal as a teenager at Highbury
'I started to drive myself crazy': Former Manchester United midfielder explains why it all went wrong for him under Erik ten Hag
