First try to fix this fucked up test