fix: missing draw func