tests: fix mock usage

Change-Id: I38bef45b09c5e90e079fcf0d4fdaf796bf79b79a