fix bug in trait