Fixed issue with selection box