fixes after test