Fix refrence issue