Minor fix about screen functions

It will run but won’t work as I suppose how it works