Fix Edit class