fix: Update stub generation helper function to handle classes followed by dataclasses

fix: Update stub generation helper function to handle classes followed by dataclasses #1024