Fix display module on edit state get extra blank

tower: https://tower.im/projects/8f19f0bf0e754f0b82ef2c24bc230973/todos/7ad19a232d75479f90ce00450f6ffe24/

Change-Id: I13a24660c2300d4e0a183f72455f2073c94b9c99