Fix deprecation

Change-Id: I8ace46aba910c14cb3c4bdeff38f2500ecef12df