Fix code climate issue