fixing naming issue on small data