diff --git a/VGSegment.podspec b/VGSegment.podspec
index 05a9aa7..5357b37 100755
--- a/VGSegment.podspec
+++ b/VGSegment.podspec
@@ -11,7 +11,7 @@ Pod::Spec.new do |s|
   s.ios.deployment_target = "8.0"
   s.platform     = :ios, "8.0"
   s.source       = { :git => "https://github.com/VeinGuo/VGSegment.git", :tag => s.version }
-  s.source_files = 'VGSegment/**/*.*'
+  s.source_files = 'VGSegment/**/*.{h,m,swift}'
   s.requires_arc = true
 
 end