04/19/2012 AT 05:45 AM EDT
Surprise! Mark Wahlberg
got a sweet visitor on set: his wife Rhea
, who stopped by to kiss the hunky actor while he's filming Pain and Gain
"Mark and Rhea enjoyed a romantic smooch in front of a few others on the set before she left, and he had some ice cream," a source tells PEOPLE. "It was cute."
Wahlberg plays a bodybuilder gone bad in the film – and he has been spending time swimming and enjoying Miami Beach with his family on weekends. – Linda Marx