Small addition to the previous fix.