fix build (thanks maxwen/omni)

Change-Id: Ic022aebcddc325b1c8fea44bf73f6e82d2e2643e