tweak ui to pass tests