Fix minor variable reference issue