Added a return;

I think you forgot that one :P