Small fix

Change-Id: I7cc8c5bb5691d8aa13319f085f2c458dc68fb266