Small fix

Change-Id: I301f273da700e65a9ca207c2233624ea47108665