Add missing assertions to solve integration test