[misc] Move class reference from the implementation to the interface