add if around block so it will not remove everything