Revert "Updated tests"

This reverts commit 59fcb7b461acad9aa5f452a2be56a71ab2926dda.