Remove channels_first argument