Fix minor bug: wrong variable