Fixed mistake from last commit