Some cleanup in aks, which is working btw.