I agree entirely with how you did this. The characters are supposed to be experienced adventurers, even if the players are not.
Usually, if a PC does something not taking into account something I think their character should know, I'll give them a check like you did. In a room, sitting on a...