When you have that idea but are distracted - try writing anyway! Just random words if need be. You can edit it later, but it'll help you to remember what you were on about...