fixes tests
fixes tests