Hacky fix for regression.