fix p_squared built-in test