Updating a fix for traits