Fixed saving code

Signed-off-by: David Evans <dgevans42@gmail.com>