diff --git a/setup.php b/setup.php
index 58146cb..4d8f873 100644
--- a/setup.php
+++ b/setup.php
@@ -112,14 +112,14 @@ function plugin_flowview_check_upgrade() {
 				$good = db_fetch_row("SHOW INDEXES
 					FROM `" . $t['TABLE_NAME'] . "`
 					WHERE Key_name='keycol'
-					AND Column_name='end_time';");
+					AND Column_name='src_if';");
 
 				if (!cacti_sizeof($good)) {
 					cacti_log('NOTE: Updating unique key for ' . $t['TABLE_NAME'], false, 'FLOWVIEW');
 
 					db_execute('ALTER TABLE ' . $t['TABLE_NAME'] . '
 						DROP INDEX `keycol`,
-						ADD UNIQUE INDEX `keycol` (`listener_id`,`src_addr`,`src_port`,`dst_addr`,`dst_port`, `start_time`, `end_time`)');
+						ADD UNIQUE INDEX `keycol` (`listener_id`, `src_addr`, `src_port`, `src_if`,`dst_addr`,`dst_port`, `dst_if`, `start_time`, `end_time`)');
 				}
 			}
 		}