Minor: Fixing bug in last update