Fix running on my machine for now