test fix
fixed tests