Fix failing register test