Minor variable fix