Add missing import when running these tests standalone.
9 files changed