Fix lil error of fixing 1 bug