diff --git a/core/src/main/scala/org/apache/spark/scheduler/StageExInfo.scala b/core/src/main/scala/org/apache/spark/scheduler/StageExInfo.scala
index 90389a45fc4267054ccc106c980e608f35afaa1e..5871f09d0c2650eab0abcc0e1453b569fb76b4d5 100644
--- a/core/src/main/scala/org/apache/spark/scheduler/StageExInfo.scala
+++ b/core/src/main/scala/org/apache/spark/scheduler/StageExInfo.scala
@@ -15,7 +15,7 @@
  * limitations under the License.
  */
 
-package org.apache.spark.storage
+package org.apache.spark.scheduler
 
 import scala.collection.mutable