Fix a couple of errors in the new tests