Quick fix for square generator mean, oops