Yes, sorry. You only need to change the next line