I just pushed a commit to use a consistent import style in both the files changed in this PR.