Fixes ui buttons on focused blocks.