Add missing import to fix build after TRDS

Change-Id: I53610d9b0f02c11a75de7fac28bf3269f06ba08a