fix reload stage