Fix accidental hard coding