fix registerRDDForCleanup