Fix small bug on fresh install