Fix register tests