add more control to the tests