Fixing return error