Screen fix after postgres integration