aks: updated for recent changes