Fixed diagram issues.