Fix data generator that was causing server to crash on startup