changed order in assertEquals
added method restore()