From b1aa636687f580ffb6263305c87fc27a145eac24 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cem=20G=C3=B6kmen?=
 <1408354+cgokmen@users.noreply.github.com>
Date: Wed, 15 Nov 2023 00:52:44 -0800
Subject: [PATCH] Fix grasp task

---
 omnigibson/tasks/__init__.py   | 1 +
 omnigibson/tasks/grasp_task.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/omnigibson/tasks/__init__.py b/omnigibson/tasks/__init__.py
index 2a6bfaa8f..550a873f0 100644
--- a/omnigibson/tasks/__init__.py
+++ b/omnigibson/tasks/__init__.py
@@ -3,3 +3,4 @@
 from omnigibson.tasks.point_navigation_task import PointNavigationTask
 from omnigibson.tasks.point_reaching_task import PointReachingTask
 from omnigibson.tasks.behavior_task import BehaviorTask
+from omnigibson.tasks.grasp_task import GraspTask
diff --git a/omnigibson/tasks/grasp_task.py b/omnigibson/tasks/grasp_task.py
index fa372cae5..f7cc448ab 100644
--- a/omnigibson/tasks/grasp_task.py
+++ b/omnigibson/tasks/grasp_task.py
@@ -1,7 +1,7 @@
 import random
 import numpy as np
 import omnigibson as og
-from omnigibson.action_primitives.starter_semantic_action_primitives import UndoableContext
+from omnigibson.action_primitives.starter_semantic_action_primitives import PlanningContext
 from omnigibson.reward_functions.grasp_reward import GraspReward
 
 from omnigibson.tasks.task_base import BaseTask