add a test
fix imports