Venue of a prosecution for an offense punishable under § 154.510 or 154.517 of this code is in Travis County or in the county where the offense occurred.