Fix error that occurred in prev commit