Fix tests for :state addition