Fixed bug on refrence