EDX-315: Fix UI

Fix UI