Fixes dumb mistakes from last commit