Fixes method signature issue