Small modifs that I forgot about