DNN-4847 Fix broken unit tests as a result of #194

When #194 was submitted, it didn't update the unit tests resulting in failed tests.  This change just fixes the failing unit test.