Fix .once() implementation