Will fix when that's merged.