Adding a missing import for previous commit.