fixes after running tests