improve error msg
fix tests