Fix to run in docker