Scott Disick received a visit from his children — 5-year-old son Mason, 3-year-old daughter Penelope and 10-month-old son Reign — while he continues treatment at a drug rehabilitation center in Malibu, Calif.

"Entertainment Tonight" reported that Kris Jenner was photographed Sunday bringing the kids to see Disick, while still dressed in striped pajamas from her daughter Kim Kardashian's baby shower earlier that day.

"ET" described Disick as being affectionate with his children, even giving Reign "a kiss on the head, as well as picking up both Penelope and Mason." The 32-year-old reality star also posed for family photos with Kris. The visit reportedly lasted two hours and "seemed fine, but a bit serious around Kris."

"He is doing better. He has cut out a lot of the bad influences in his life, and is trying to focus on getting back to normal," a source close to the family told People. "He's still got the demons; they haven't gone away. But now he recognizes that he has those demons, which is a big step for Scott. He hasn't acknowledged his struggles before, not in a meaningful way. He's always just wanted to hold on and hope that he'll do okay. Now he's digging deeper."

Added the source: "He's in a different place. The question is whether this is a permanent change or something temporary. Only Scott can answer that."

Disick entered rehab shortly after Lamar Odom was found unresponsive in Las Vegas after doing cocaine and taking sexual performance-enhancement drugs. A source told Us Weekly that Odom's "condition is really hitting Scott hard."

People reported that since ending their nine-year relationship, Disick and Kourtney Kardashian have been in a custody battle. However, a source told "ET" in September that Kardashian "has been adamant about not keeping their children away from their father."