fix bug annotating registers