Update mthod calls in trait