Add import in decoration