Fix Entity action bug