Update selection directive to handle cleared model state