Keep type arguments of = when unelaborating

This allows more informative error messages when the mismatch is in the
type arguments to =.