fix: use appropriate input_type