omega_sq bug fix