Fix missing stuff