Fixes calculate_snapshot_restore_timeout() method.

Change-Id: Ia81a4b0572475767b010b97e9876744ac5f24c38