Fixed registration
small fix