Fixing the fails I missed