Fixed up the reload function