Mario Ballotelli to AC Milan: Maverick striker to have 'good behaviour' clause inserted in Milan contract as he completes loan from Liverpool

Italian is on the brink of finalising return to Milan

Mario Balotelli is all smiles at AC Milan's training ground as he closes in on switch (twitter.com/acmilan)

No more setting off fireworks or smoking then.

Mario Balotelli will have a good behaviour clause inserted into his contract with AC Milan, banning extravagant haircuts and clothing and committing him to a healthy lifestyle.

Italian sports daily Gazzetta dello Sport have claimed that the club's chief executive Adriano Galliani had based the clause on regulations which are applied to people who serve in the Italian air force.

The 25-year-old, repeatedly criticised for his poor work rate, is set to return to Milan just one year after leaving for Liverpool, where he had a dismal season.

He underwent a medical in Milan on Tuesday ahead of an unexpected loan move, although the deal has not yet been officially confirmed.

According to Gazzetta, Balotelli must not damage the image of the club and his Twitter, Facebook and Instagram posts will be closely monitored.

He will be banned from extravagant haircuts and clothing, from smoking and from visiting night clubs and must turn up on time from training. His drinking will also be limited.

Balotelli's first spell at Milan was relatively successful and uneventful with a respectable scoring rate although there were some misdemeanors.

He lost his temper in front of the cameras after his performance in was criticised by Italian television pundits, telling his interviewers that they did not understand football and throwing his microphone down.

He was also booked for making a vulgar gesture at Cagliari supporters, served two three-match bans and was caught smoking in the toilet of a train by a ticket collector.

However, these incidents paled in comparison to some of his antics at Manchester City earlier in his career, where he famously set fire to his house after letting off fireworks in his bathroom.

Additional reporting by Reuters