Adds a couple of assertions to tests