UI final fix