[Unit Tests] Fix error in last Unit Tests

* Oops