From eb0aed7c4b15c37c46b15b5d9e32ceb93f2e1e95 Mon Sep 17 00:00:00 2001
From: Peter LaFosse <peter@vector35.com>
Date: Tue, 1 Sep 2020 12:43:10 -0400
Subject: Add support for custom flowgraph edges

---
 python/examples/triage/entropy.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'python/examples/triage/entropy.py')

diff --git a/python/examples/triage/entropy.py b/python/examples/triage/entropy.py
index 57e9d2c4..3ce2de11 100644
--- a/python/examples/triage/entropy.py
+++ b/python/examples/triage/entropy.py
@@ -4,8 +4,8 @@ from PySide2.QtWidgets import QWidget
 from PySide2.QtGui import QImage, QColor, QPainter
 from PySide2.QtCore import Qt, QSize, QTimer
 import binaryninjaui
-from binaryninjaui import ViewFrame, ThemeColor, UIContext
-
+from binaryninjaui import ViewFrame, UIContext
+from binaryninja.enums import ThemeColor
 
 class EntropyThread(threading.Thread):
 	def __init__(self, data, image, block_size):
-- 
cgit v1.3.1