fix ones attention