added more checks in test