small fix
fixing some tests