Fixing ensure

Change-Id: I4ea7baa53af6fbe5366ae6b9d786e3315afbb08c