Fixed errors when running Rubocop