fix a import error on CI - just made a comment