Fixing build broken by this change

c43ca06ca091fc09e2c86bb051ac92b648f12b64