It's not illegal to fire someone who is pregnant. It's illegal to fire someone because they're pregnant. You have to be able to point to another reason, such as performance.

Meaning, if you're doing layoffs, and you thought to yourself, "Let's fire person X! They're about to have a kid and won't even be around for two months. That'd be a waste of money!", you would be breaking the law.