Two little changes I forgot about