Accidentally deleted a return statement. Fixed now.