nit: seems like this assert is not needed, can we just remove it?