#compdef source

if [[ CURRENT -ge 3 ]]; then
  compset -n 2
  _normal
else
  _files
fi