Fixing deprecated reference