fixes tests that were previosuly breaking