Balotelli is serving a three-match ban for confronting match officials, but is available for their Champions League clash with Ajax.
Allegri said: "Tomorrow we'll have Balotelli back, but I think his recent suspension has shown him that he has to change his attitude on the pitch.
"In Europe there are no easy games. Ajax play some good football and have a great tradition. We'll need to be very careful."
For breaking stories and all the great banter like us on Facebook: facebook.com/tribalfootball