No need for elseif with returns :)

/cc @keradus