Add test asserts