Fix API bug