fix friggin tests