Updated tests

Correction to the latest commit