Fix completion script