fixing some stuff test case