Fix: refactored method signature