Little fixes to warning