Fixed for production
A small ui fix