[FIX] Fixed reference bug