Update test to use update-state!