Fix old API hack