Fixed a reference bug