Fixes failing restaurant test