Revert "Move that import in case it helps jenkins"

This reverts commit 94c03df89d60bab350a4fca35fc2cd3cb4b7e451.
Add back filename