@@ -14,6 +14,8 @@ import Button from './base/forms/Button'
1414import Tooltip from './Tooltip'
1515import { withRouter } from 'react-router-dom'
1616import { getDarkMode } from 'project/darkMode'
17+ import { getStore } from 'common/store'
18+ import { removeProjectFlag } from 'common/services/useProjectFlag'
1719
1820const featureNameWidth = 300
1921
@@ -207,7 +209,7 @@ class CompareEnvironments extends Component {
207209
208210 { this . state . environmentLeft && this . state . environmentRight ? (
209211 < FeatureListProvider onSave = { this . onSave } onError = { this . onError } >
210- { ( { } , { removeFlag , toggleFlag } ) => {
212+ { ( { } , { toggleFlag } ) => {
211213 // Adapt old FeatureListProvider signatures to new FeatureRow signatures
212214 const adaptedToggleFlag =
213215 ( environmentId ) => ( projectFlag , environmentFlag , onError ) => {
@@ -220,7 +222,10 @@ class CompareEnvironments extends Component {
220222 )
221223 }
222224 const adaptedRemoveFlag = ( projectFlag ) => {
223- removeFlag ( this . props . projectId , projectFlag )
225+ removeProjectFlag ( getStore ( ) , {
226+ flag_id : projectFlag . id ,
227+ project_id : this . props . projectId ,
228+ } )
224229 }
225230 const renderRow = ( p , i , fadeEnabled , fadeValue ) => {
226231 const environmentLeft = ProjectStore . getEnvironment (
0 commit comments