Adds registration back in