Make it work with ActionSheet