Small fix to make this setup work right now.