Fix: missing return with hook