fix tests
Set figsize to (10,10)