Fixed some missed referrences