W wyborach parlamentarnych w 2015 wystartował do Sejmu w okręgu toruńskim z pierwszego miejsca na liście wyborczej PO. Uzyskał mandat posła VIII kadencji, otrzymując 13 400 głosów. Został wybrany na przedstawiciela Sejmu w Krajowej Radzie ProkuraturyUchwała Sejmu Rzeczypospolitej Polskiej z dnia 25 listopada 2015 r. w sprawie wyboru posła członka Krajowej Rady Prokuratury (). (zlikwidowanej w 2016). W Sejmie VIII kadencji został członkiem Komisji Sprawiedliwości i Praw Człowieka, Komisji Ustawodawczej oraz Komisji Nadzwyczajnej do spraw zmian w kodyfikacjach, pracował też w Komisji Gospodarki i Rozwoju (2015–2016).
W wyborach w 2019 z powodzeniem ubiegał się o poselską reelekcję z ramienia Koalicji Obywatelskiej, otrzymując 32 439 głosów. W Sejmie IX kadencji został wybrany na przewodniczącego Komisji Ustawodawczej.
W wyborach w 2023 po raz kolejny uzyskał mandat poselski (otrzymał 64 452 głosy). Objął następnie funkcję przewodniczącego Komisji Sprawiedliwości i Praw Człowieka. W grudniu tego samego roku został powołany na sekretarza stanu w Ministerstwie Sprawiedliwości. W styczniu 2024 został powołany na członka i przewodniczącego Komisji do spraw reprywatyzacji nieruchomości warszawskich.