SPARK-2277: remove unnecessary import