Phelps , who was banned for three months when he was pictured smoking drugs at a party after returning from Beijing , has been eager to regain his Olympic form after his enforced exile from the pool .