Fix Tests

One too many returns got deleted...