fixing mistake the gf pointed out