Fixed failing unit tests due to not being updated from a previous update