Fix build break from theanax