diff --git a/torchtrain/profiling.py b/torchtrain/profiling.py
index d7609438..167067e0 100644
--- a/torchtrain/profiling.py
+++ b/torchtrain/profiling.py
@@ -1,8 +1,9 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
+# This software may be used and distributed according to the terms of the Llama 2 Community License Agreement.
 
 import contextlib
 import os
+
 import torch
 
 try: