Nicole Johnson Celebrates Michael Phelps' Final Swim with Baby Boomer and Parents

In Rio, Phelps added 6 medals – 5 gold and 1 silver – to his astonishing tally of 28 Olympic medals

Photo: Nicole Johnson/Instagram

Nicole Johnson teared up as Michael Phelps won gold in his final Olympic race, but she couldn’t be happier with her fiancé’s performance at the Rio Games.

The former Miss California, 31, shared a photo with her parents, Phelps’ mother Debbie and her 3-month-old son Boomer reflecting on Phelps’ historic week in Rio.

“It’s been an incredible week with the family cheering on @m_phelps00,” Johnson captioned the snap, adding that she wished Phelps’ sisters were able to join.

In Rio, Phelps added 6 medals – 5 gold and 1 silver – to his astonishing tally of 28 Olympic medals. For each race, Johnson and the couple’s son (wearing his signature red, white and blue headphones) were watching from the stands.

For more of PEOPLE’s Olympic coverage, pick up our collector’s edition, The Best of the Games, on sale now.

Road to Rio: How Moms Support Olympic Hopefuls

The 31-year-old swimmer has repeatedly said that these are his last Olympic games – a sentiment that he repeated in his post-race interview.

“I’m just ready for something different,” he told NBC. “My swimming career might be over, but I have the future ahead of me. It’s not the end of a career, [it’s] the beginning of a new journey.”

That new journey includes a wedding, which Johnson revealed during the Games had a date set.

“We have a date,” Johnson told Access Hollywood. “It’ll be small and intimate for the wedding, and then we’re throwing a massive bash for everyone in the states.”

She added, “We’re kinda getting things rolling, I’ve been planning as we’ve been leading up to [the Olympics]. So I think I have the wedding in order, and now it’s on to the party for the fun.”