BUG: Fixed axes issue in eda