Hook the correct class