Just realised an error here - we don't want to return the translated context label, but the extension type.

if ( isset( $context_labels[ $ext_type ] ) ) return $ext_type;

return $ext_type;

Make sense?