Code Climate fix