More piano roll fixes