Review changes made to tests