Wolverhampton Wanderers manager Bruno Lage believes his team needs to improve on killing off games after they held on for a 2-1 win over Aston Villa.