Update tests

Fix indent

fix tests