write test
fix