forgot to update this into the function