Move assert into the test