fix type
fixed tests