Add assertions to Unit