Fix episodic memory when following plans