small fix for previous fix