Fix to make Piazza work

Fix to make Piazza work