Fix compilation

Change-Id: Idf445b94c1b43ab4f4c3b2e121134f7a722be1d6
Reviewed-by: Orgad Shaneh <orgads@gmail.com>