Fix something I forgot