Correctly document return types for ast nodes