Fixed the bad tests