Quick fix to a bug that accidentially got introduced from last commit