fix to run tests properly