discord python wait for user input

1await ctx.send(f'Role {role.mention} was set as default role. Do you want to apply this to all users in the server? [YES/NO]')
2	def check(m):
3    	return m.channel == ctx.channel
4    msg = await bot.wait_for('message', check=check, timeout=500)
5    checkcontinue = msg.content.lower()
6    if checkcontinue == 'yes':
7      await ctx.send(f'Assigning roles... This could take a little while!')