Also add this forgotten part

I need sleep. pfff