Fix for change of method signature