Fix reference to method that doesn't exist in the Grid module