Fixed a test
Fixed another test