Fixed the compilation error because of Spark's unpersist API change.