Fixed wrong UI