Fix code climate