Fixed mistake from previous commits