Fixes tests….somehow