Minor update to get it working again.