Fix codes class import