Fixes after rubocop