A training device for training a user on the operation of an auto-injector that dispenses a medicament. A needle guard is slidable from a first extended position relative to a housing prior to operation of the training device, to a retracted position during operation of the training device, and to a second extended position after operation of the training device. An actuation assembly controls the movement of the needle guard from the retracted position to the second extended position. A lock locks the needle guard in the second extended position. A safety is proximate a proximal end of the housing and prevents movement of the needle guard from the first extended position to the retracted position in a storage position. The needle guard is moveable into the retracted position with the safety in an armed position. The safety resets the lock upon returning the safety to the storage position.