Fixed bugs that some prompt cannot edit

The cause of the bug is a single missing line of code