Also, mark this and the class above as `private[spark]`.