Change height to original after submit
Fixed missing import