Updated tests.

Updated tests.