Update Generator tests