um: Fix fpstate handling