Added check on head, missed this before