Fix test that failed because new classes were introduced.

Refs #3689