Nothing needs to be changed besides a line you mentioned.