handle failures in test better