Update some tests