Add missing import when running these tests standalone.
1 file changed