Small fix

fixes things I broke