Modified test to tests