fixed bug

restore problem fixed