Fix trial bug which caused by interface change