Small update to the hook. Waiting on the API to consume.