Bug fix

Fixed issue caused by addition of second grid