Added possibility to use the proper screenshot