Fix dirty bug
Dirty fix