Fix omega to squared