Ray Kroc, the onetime milkshake machine salesman who made McDonald’s into a global fast-food icon, would be proud.

»MORE: Publix deal alert: Pub Subs to go on sale for $5.99 next week

Some McDonald’s locations are getting a device that will help keep the chain’s ice cream machines from breaking down all the time. And, it’s just in time for Shamrock Shake season.

Kytch, a California-based software company, has created a device that can correct any minor malfunctions and let workers know when machines are breaking down, according to Business Insider.

McDonald’s ice cream machines feature an automated cleaning cycle that takes up to four hours to complete.

Kytch’s new device, which launched in May 2019, will be able to anticipate if part of the machine is about to break and warn employees.

The upgrades come right as McDonald’s returns its Shamrock Shake on Feb. 19, marking the treat’s 50th anniversary. McDonald’s is taking the Shamrock Shake nationwide for the first time since 2017 and also is introducing the new OREO Shamrock McFlurry for a limited time.

The Shamrock Shake is a vanilla soft serve blended with McDonald’s Shamrock flavor and topped with a whipped topping.

The new OREO Shamrock McFlurry is a vanilla soft serve with a Shamrock flavor and OREO cookie pieces blended.