Fix for missed argument