fixes tests (omg ;)