The mentalist - what do to when one of the cast gets pregnant

Anyone else been following this on The Mentalist. Grace, one of the team of detectives, is obviously pregnant. I spotted this many episodes back, and as the serious continues she gets more and more obviously so. However the plot does not include this pregnancy. So first we see her desk bound (rather than running after baddies) and when standing, she is always holding a file and wearing a loose jacket. Most recently she is only shown head and shoulders. Must be a real challenge for the directors and writers - but quite amusing to watch.