Hack to make it work in i3. (will be redone better)