Fixed bug with multiple returns

Didn't notice the multiple returns