Fixes main-class attribute
Fixes an error