Better assertions in test