Update component to match the latest engine API