Fix edit works