Fix operator=() to return reference