Fix restore selection
Working, need to change paths