Fix drag bug