Fixed regression issue with last change.