Fixed remnant from older version of code