update example
fix tests