Fixed registration controller issue