Fixed failing unit test
Revert "Fixed failing unit test"

This reverts commit df77f1fe41f824f6f009770b15acd1b5d1a3cf13.
Fixed failing unit test