Aubrey Plaza and Jeff Baena separated months before the screenwriter passed away