Use correct state for output