Small fix for previous fix