make ai module works again