Fix omega bug