Small refactoring on screenshot [JacoPretorius]