Avril Lavigne keeps her head well above water during return