Fix damn tests