Fix trait to match interface