Fixed an issue with registration