It is not so difficult to include such a scene ib HBP: A Party at the Slug-Club.
But Slughorn has mixed up the things a little bit (or anyone else) So they drink Polyjuice Potion instead of whatever.