Change to better fit with rubocop conventions