assign measurement for the case of 1 state

I believe the function is saner this way