Fixed bug for getting selection model
Fixing selection bug