Fix registration check for HSA