Fixing return values