Fixup import syntax error in unit test

Change-Id: I12f4d94474a31d25819fe24b70676f81c52bd022