OrchardCMS/OrchardCore

I have a controller that creates a shape like this

public async Task<IActionResult> Display()
{
    Location location = new Location()
    {
        Name = "Test 1",
    }

    IShape shape = await _displayManager.BuildDisplayAsync(location, this);

    return View(shape);
}

Here is how my location driver looks like LocationDisplayDriver

public class LocationDisplayDriver : DisplayDriver<Location>
{

    public override IDisplayResult Display(Location model, IUpdateModel updater)
    {
        return Initialize<DisplayLocationViewModel>(nameof(DisplayLocationViewModel), viewModel =>
        {
            viewModel.Id = model.Id;
            viewModel.Name = model.Name;
        }).Location("Content");
    }
}

I created a view called ~/Views/DisplayLocationViewModel.cshtml while the Locations/Display.cshtml view has @await DisplayAsync(Model)

But when I try to render the shape using @await DisplayAsync(Model) I get this error

Exception: Shape type 'Location' not found

I am expecting the shape to be rendered using ~/Views/DisplayLocationViewModel.cshtml view. What am I missing here that is causing this error?