Fixing missing assignment operator