Fix for running locally