fix. Ensure args are provided in tests where required.

Change-Id: Ie08016bdffcf1d8a33b6435e34cc5a9ecbb6fcb5