Fix to registration test