Add type for channel argument