#336 fix
fixed tests