Modified coverage to work with rails