Fix what I forgot to remove