Fixes mistake from last commit