small bug related with the previous change