While I appreciate the advice you have been given, I find it odd that such a 'government policy' has nothing in writing! Everything that workplaces follow needs to be documented in some way shape size or form, so I find it odd*for HR to say there was no such written advice.
Regardlesss, I hope you and baby are well, and you might as well make the most of the time!! Every cloud and all that!