Fix trait extending bug