Revert "Updated tests"

This reverts commit c30340fd0cb4094e5b28ae777bdd532c7f7c3d87.