Fix potential class cast exception during selection handling