Custom-submit directive fix