Add 'return' that we actually need