diff --git a/.travis.yml b/.travis.yml
index 5e6c44b..58157cd 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -10,3 +10,5 @@ before_script:
 script:
   - bundle exec rake test
   - bundle exec rake build
+after_script:
+  - '[ ${TRAVIS_EVENT_TYPE} != "pull_request" ] && [ ${TRAVIS_BRANCH} = "master" ] && bundle exec codeclimate-test-reporter'
diff --git a/lib/fluent/plugin/in_dynamodb_streams.rb b/lib/fluent/plugin/in_dynamodb_streams.rb
index a2f7ccf..54693b1 100644
--- a/lib/fluent/plugin/in_dynamodb_streams.rb
+++ b/lib/fluent/plugin/in_dynamodb_streams.rb
@@ -1,3 +1,4 @@
+require 'fluent/input'
 module Fluent
   class DynamoDBStreamsInput < Input
     Fluent::Plugin.register_input('dynamodb_streams', self)
diff --git a/test/test_helper.rb b/test/test_helper.rb
index 5fa861d..8f66af5 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -1,6 +1,3 @@
-require "codeclimate-test-reporter"
-CodeClimate::TestReporter.start
-
 require 'fluent/test'
 require 'fluent/plugin/in_dynamodb_streams'
 require 'aws-sdk'