fixed this functionality