Add missing assertion