moves return; to correct location

I accidently put the return; inside the else block I added