Fixing specs
Changes from Rubocop