Setting menu was not doing anything so fixed that. Tests pass.