Set correct type for param union