Fix a small bug (forgot to change the variable names...)