[FIX] Fixed variable not set error