Added argument

Not sure why it wasn't there after I pulled.