Adding back a missing parameter

Change-Id: I3f8d7fe716fa080bd46ce24c1fc75bfea72b3b90