Minor update to our tests