Address review comments + better errors for graph extension failure