fixing one diagonal case