Fixed Entity Registration Bug