Accidentally left out some files -- fixing this bug