fixing some test