Added couple of missing lines

Change-Id: I34ad77aac0c075eb0bcb3a85ac9204c57f5af13d