Fixing a little hard coding error