added more checks in the test