Fix assertion 2

Change-Id: I31e7c0f1e1115a02d93964c719be5fcbb5ab81fb