diff --git a/spec/defines/file_spec.rb b/spec/defines/file_spec.rb
index 4624849..db10fc0 100644
--- a/spec/defines/file_spec.rb
+++ b/spec/defines/file_spec.rb
@@ -1,65 +1,66 @@
 require 'spec_helper'
 
 describe '::cassandra::file' do
   let(:pre_condition) do
     [
       'class stdlib () {}',
       'define ini_setting($ensure = nil,
          $path,
          $section,
          $key_val_separator       = nil,
          $setting,
          $value                   = nil) {}',
       'define file_line($line, $path, $match) {}'
     ]
   end
 
   let!(:stdlib_stubs) do
     MockFunction.new('concat') do |f|
       f.stubbed.with([], '/etc/cassandra')
        .returns(['/etc/cassandra'])
       f.stubbed.with([], '/etc/cassandra/default.conf')
        .returns(['/etc/cassandra/default.conf'])
       f.stubbed.with(['/etc/cassandra'], '/etc/cassandra/default.conf')
        .returns(['/etc/cassandra', '/etc/cassandra/default.conf'])
     end
     MockFunction.new('strftime') do |f|
       f.stubbed.with('/var/lib/cassandra-%F')
        .returns('/var/lib/cassandra-YYYY-MM-DD')
     end
   end
 
   context 'On a Debian OS set the max and new heap size' do
     let :facts do
       {
         osfamily: 'Debian'
       }
     end
 
     let(:title) { 'cassandra-env.sh' }
 
     let :params do
       {
         config_path: '/etc/cassandra',
         'file_lines' => {
           'MAX_HEAP_SIZE 4GB' => {
             'line'  => 'MAX_HEAP_SIZE="4G"',
             'match' => '^#MAX_HEAP_SIZE="4G"$'
           }
         }
       }
     end
 
     it do
       should contain_class('cassandra')
       should contain_class('cassandra::params')
       should contain_class('stdlib')
+      should contain_cassandra__file('cassandra-env.sh')
 
       should contain_file_line('MAX_HEAP_SIZE 4GB').with(
         path: '/etc/cassandra/cassandra-env.sh',
         line: 'MAX_HEAP_SIZE="4G"',
         match: '^#MAX_HEAP_SIZE="4G"$'
       )
     end
   end
 end