Fixed mistake from previoius commit