Only import modules in tools/flakes

Change-Id: I049f4e4d3927d0bf792697d30250083c87ca0b73
Merge "Only import modules in tools/flakes"