Fixes up previous commit