The Grammy Award winner Rihanna displayed her baby bump on Saturday as she stepped out with Rocky, 33, for what one source called a “happy and relaxed” dinner with an intimate group of friends at Giorgio Baldi in Santa Monica, California, following Rocky’s arrest earlier that week.

“It was a happy and relaxed group. Rihanna looked great. She sat next to A$AP and they were affectionate,” the source said adding: “They dined for a couple of hours. The focus was very much on Rihanna and the baby.”

This was after Rocky was arrested on Wednesday at Los Angeles International Airport in connection with a shooting that took place in November. He was detained by the Los Angeles Police Department after he arrived on a private plane that morning, returning from a vacation with Rihanna in her native Barbados.

The artist was charged with assault with a deadly weapon (firearm), which is a felony, and his bail was set at $550,000. Rocky was released from jail later that day.

A$AP Rocky and Rihanna

He allegedly shot a man following an argument, and the victim claimed that Rocky fled the scene on foot with two other men. The case will be presented to the Los Angeles County District Attorney’s Office for filing consideration, according to police.

A rep for Rocky didn’t respond to people’s requests for comment after the arrest. He has not appeared in court in connection with the allegations.

Rihanna and Rocky announced their pregnancy in January when the mother-to-be revealed her baby bump as they were photographed together during an outing in New York City.

Rihanna said it’s fun” but also “a challenge” to play with fashion while pregnant. “I like it. I’m enjoying it,” she said in February. “I’m enjoying not having to worry about covering up my tummy,” Rihanna added then. “If I feel a little chubby, it’s like, whatever! It’s a baby!”