Fix missing argument

Change-Id: Ie47221444968fa5850ff637c1a28fc8cb034bf2f