Fixed mistake from previous commit