Fixed something we forgot