Fix unit tests.

Refs #5986