Fixed old code class import.