minor fix to axi types