diff --git a/lib/ajax-datatables-rails.rb b/lib/ajax-datatables-rails.rb
index 0965af97..a3491cc0 100644
--- a/lib/ajax-datatables-rails.rb
+++ b/lib/ajax-datatables-rails.rb
@@ -1,11 +1,11 @@
 # require 'rails'
 
 class AjaxDatatablesRails
-  
+
   class MethodError < StandardError; end
 
   VERSION = '0.0.1'
-    
+
   attr_reader :columns, :model_name, :searchable_columns
 
   def initialize(view)
@@ -19,7 +19,7 @@ def method_missing(meth, *args, &block)
   def as_json(options = {})
     {
       sEcho: params[:sEcho].to_i,
-      iTotalRecords: @model_name.count,
+      iTotalRecords: get_raw_record_count,
       iTotalDisplayRecords: filtered_record_count,
       aaData: data
     }
@@ -38,7 +38,7 @@ def get_raw_records
   def filtered_record_count
     search_records(get_raw_records).count
   end
-  
+
   def fetch_records
     search_records(sort_records(paginate_records(get_raw_records)))
   end