Fix input types from D13050

Summary: This differential fixes types according to Ashoat suggestion: https://phab.comm.dev/D13050#inline-76348

Test Plan: flow

Reviewers: ashoat

Reviewed By: ashoat

Subscribers: tomek

Differential Revision: https://phab.comm.dev/D13271