Fix: code cause crash