Fixed referencing issue