Added missing import for unittest

Change-Id: I6a666557bf02aa48d4c0d58273fe0744fe91b888
Merge "Added missing import for unittest"