Fix internal module use