Fix assertion

Change-Id: Id98bc37ddddda30fa91cdc427931c279e3a93355