Fixed something I missed before