AEI fixes for when aei matrix passed in has only one tag snp