Fix 'length of undefined' error