Use correct parameter

Change-Id: I8c56645ce70f17118f5926e5f0e2edc88f9e2181