Fix bug failing code on staging