fix the import for trial