fix #5 for traits